Spaces:
Runtime error
Runtime error
update app.py
Browse files
app.py
CHANGED
|
@@ -17,11 +17,11 @@ from huggingface_hub import hf_hub_download
|
|
| 17 |
converter = opencc.OpenCC('s2t.json')
|
| 18 |
|
| 19 |
# 加载模型
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
cosyvoice_zjg = CosyVoice2(
|
| 26 |
'ASLP-lab/Cosyvoice2-Yue-ZoengJyutGaai',
|
| 27 |
load_jit=False, load_trt=False, load_vllm=False, fp16=False
|
|
@@ -39,10 +39,10 @@ def tts_inference(model_choice, text, prompt_audio):
|
|
| 39 |
model = cosyvoice_zjg
|
| 40 |
prompt_audio = "asset/sg_017_090.wav"
|
| 41 |
elif model_choice == "CosyVoice2-精品女音":
|
| 42 |
-
model =
|
| 43 |
prompt_audio = "asset/F01_中立_20054.wav"
|
| 44 |
elif model_choice == "CosyVoice2-base":
|
| 45 |
-
model =
|
| 46 |
if prompt_audio is None:
|
| 47 |
return None, "请上传参考音频"
|
| 48 |
else:
|
|
|
|
| 17 |
converter = opencc.OpenCC('s2t.json')
|
| 18 |
|
| 19 |
# 加载模型
|
| 20 |
+
cosyvoice_base = CosyVoice2(
|
| 21 |
+
'ASLP-lab/Cosyvoice2-Yue',
|
| 22 |
+
load_jit=False, load_trt=False, load_vllm=False, fp16=False
|
| 23 |
+
)
|
| 24 |
+
print('load model 1')
|
| 25 |
cosyvoice_zjg = CosyVoice2(
|
| 26 |
'ASLP-lab/Cosyvoice2-Yue-ZoengJyutGaai',
|
| 27 |
load_jit=False, load_trt=False, load_vllm=False, fp16=False
|
|
|
|
| 39 |
model = cosyvoice_zjg
|
| 40 |
prompt_audio = "asset/sg_017_090.wav"
|
| 41 |
elif model_choice == "CosyVoice2-精品女音":
|
| 42 |
+
model = cosyvoice_base
|
| 43 |
prompt_audio = "asset/F01_中立_20054.wav"
|
| 44 |
elif model_choice == "CosyVoice2-base":
|
| 45 |
+
model = cosyvoice_base
|
| 46 |
if prompt_audio is None:
|
| 47 |
return None, "请上传参考音频"
|
| 48 |
else:
|