Spaces:
Running
on
Zero
Running
on
Zero
update app.py
Browse files
app.py
CHANGED
|
@@ -17,16 +17,16 @@ from huggingface_hub import hf_hub_download
|
|
| 17 |
converter = opencc.OpenCC('s2t.json')
|
| 18 |
|
| 19 |
# 加载模型
|
| 20 |
-
cosyvoice_base = CosyVoice2(
|
| 21 |
-
|
| 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
|
| 28 |
# )
|
| 29 |
-
# print('load model
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
# cosyvoice_biaobei = CosyVoice2(
|
| 31 |
# 'pretrained_models/CosyVoice2-yue-biaobei',
|
| 32 |
# load_jit=False, load_trt=False, load_vllm=False, fp16=False
|
|
@@ -36,13 +36,13 @@ print('load model 1')
|
|
| 36 |
def tts_inference(model_choice, text, prompt_audio):
|
| 37 |
# 选择模型和默认音频
|
| 38 |
if model_choice == "CosyVoice2-张悦楷粤语评书":
|
| 39 |
-
model =
|
| 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
|
| 28 |
+
)
|
| 29 |
+
print('load model 2')
|
| 30 |
# cosyvoice_biaobei = CosyVoice2(
|
| 31 |
# 'pretrained_models/CosyVoice2-yue-biaobei',
|
| 32 |
# load_jit=False, load_trt=False, load_vllm=False, fp16=False
|
|
|
|
| 36 |
def tts_inference(model_choice, text, prompt_audio):
|
| 37 |
# 选择模型和默认音频
|
| 38 |
if model_choice == "CosyVoice2-张悦楷粤语评书":
|
| 39 |
+
model = cosyvoice_zjg
|
| 40 |
prompt_audio = "asset/sg_017_090.wav"
|
| 41 |
elif model_choice == "CosyVoice2-精品女音":
|
| 42 |
+
model = cosyvoice_zjg
|
| 43 |
prompt_audio = "asset/F01_中立_20054.wav"
|
| 44 |
elif model_choice == "CosyVoice2-base":
|
| 45 |
+
model = cosyvoice_zjg
|
| 46 |
if prompt_audio is None:
|
| 47 |
return None, "请上传参考音频"
|
| 48 |
else:
|