gz412 commited on
Commit
4f334a8
·
1 Parent(s): c978ac1

fix requirments

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -9,6 +9,8 @@ sys.path.append('third_party/Matcha-TTS')
9
  from cosyvoice.cli.cosyvoice import CosyVoice2
10
  from cosyvoice.utils.file_utils import load_wav
11
 
 
 
12
  # 繁简转换
13
  converter = opencc.OpenCC('s2t.json')
14
 
@@ -30,10 +32,10 @@ def tts_inference(model_choice, text, prompt_audio):
30
  # 选择模型和默认音频
31
  if model_choice == "CosyVoice2-张悦楷粤语评书":
32
  model = cosyvoice_zjg
33
- prompt_audio = "asset/default_zjg.wav"
34
  elif model_choice == "CosyVoice2-精品女音":
35
  model = cosyvoice_base
36
- prompt_audio = "asset/default_biaobei.wav"
37
  elif model_choice == "CosyVoice2-base":
38
  model = cosyvoice_base
39
  if prompt_audio is None:
@@ -65,7 +67,8 @@ demo = gr.Interface(
65
  fn=tts_inference,
66
  inputs=[
67
  gr.Dropdown(
68
- ["CosyVoice2-base", "CosyVoice2-张悦楷粤语评书", "CosyVoice2-精品女音"],
 
69
  label="选择模型", value="CosyVoice2-base"
70
  ),
71
  gr.Textbox(lines=2, label="输入文本"),
 
9
  from cosyvoice.cli.cosyvoice import CosyVoice2
10
  from cosyvoice.utils.file_utils import load_wav
11
 
12
+ from huggingface_hub import hf_hub_download
13
+
14
  # 繁简转换
15
  converter = opencc.OpenCC('s2t.json')
16
 
 
32
  # 选择模型和默认音频
33
  if model_choice == "CosyVoice2-张悦楷粤语评书":
34
  model = cosyvoice_zjg
35
+ prompt_audio = "asset/sg_017_090.wav"
36
  elif model_choice == "CosyVoice2-精品女音":
37
  model = cosyvoice_base
38
+ prompt_audio = "asset/F01_中立_20054.wav"
39
  elif model_choice == "CosyVoice2-base":
40
  model = cosyvoice_base
41
  if prompt_audio is None:
 
67
  fn=tts_inference,
68
  inputs=[
69
  gr.Dropdown(
70
+ ["CosyVoice2-base", "CosyVoice2-张悦楷粤语评书"],
71
+ # ["CosyVoice2-base", "CosyVoice2-张悦楷粤语评书", "CosyVoice2-精品女音"],
72
  label="选择模型", value="CosyVoice2-base"
73
  ),
74
  gr.Textbox(lines=2, label="输入文本"),