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

update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -11
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
- '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,13 +36,13 @@ print('load model 1')
36
  def tts_inference(model_choice, text, prompt_audio):
37
  # 选择模型和默认音频
38
  if model_choice == "CosyVoice2-张悦楷粤语评书":
39
- model = cosyvoice_base
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:
 
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: