coraKong commited on
Commit
bc662b1
·
1 Parent(s): 57b4b66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -2,12 +2,13 @@ import gradio as gr
2
  from TTS.api import TTS
3
 
4
  # List available 🐸TTS models and choose the first one
5
- model_name = TTS.list_models()[0]
6
  # Init TTS
7
  tts = TTS(model_name, progress_bar=False, gpu=False)
8
 
9
  def text_to_speech(text: str, speaker_wav, language: str, model):
10
- tts = TTS(model_name, progress_bar=False, gpu=False)
 
11
  if speaker_wav:
12
  tts.tts_to_file(text, speaker_wav=speaker_wav, language=language, file_path="output.wav")
13
  else:
 
2
  from TTS.api import TTS
3
 
4
  # List available 🐸TTS models and choose the first one
5
+ model_name = "tts_models/zh-CN/baker/tacotron2-DDC-GST"
6
  # Init TTS
7
  tts = TTS(model_name, progress_bar=False, gpu=False)
8
 
9
  def text_to_speech(text: str, speaker_wav, language: str, model):
10
+ if model != model_name:
11
+ tts = TTS(model, progress_bar=False, gpu=False)
12
  if speaker_wav:
13
  tts.tts_to_file(text, speaker_wav=speaker_wav, language=language, file_path="output.wav")
14
  else: