gz412 commited on
Commit
7238dfb
·
1 Parent(s): 3898757

update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -17,11 +17,11 @@ from huggingface_hub import hf_hub_download
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,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 = 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:
 
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: