SohomToom commited on
Commit
bf698fd
·
verified ·
1 Parent(s): 8ff2cdf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -42,6 +42,7 @@ def load_tts_model(model_key):
42
 
43
  def extract_speakers(model_key):
44
  info = VOICE_MODELS[model_key]
 
45
  if info["model_name"] == "tts_models/en/vctk/vits":
46
  # Common VCTK speakers
47
  return ["p225", "p226", "p227", "p228", "p229", "p230", "p231", "p232", "p233", "p234"]
@@ -51,6 +52,7 @@ def extract_speakers(model_key):
51
  return []
52
 
53
 
 
54
  def docx_to_wav(doc_file, selected_voice, selected_speaker=None):
55
  info = VOICE_MODELS[selected_voice]
56
  tts = load_tts_model(selected_voice)
 
42
 
43
  def extract_speakers(model_key):
44
  info = VOICE_MODELS[model_key]
45
+ if info["multi_speaker"]:
46
  if info["model_name"] == "tts_models/en/vctk/vits":
47
  # Common VCTK speakers
48
  return ["p225", "p226", "p227", "p228", "p229", "p230", "p231", "p232", "p233", "p234"]
 
52
  return []
53
 
54
 
55
+
56
  def docx_to_wav(doc_file, selected_voice, selected_speaker=None):
57
  info = VOICE_MODELS[selected_voice]
58
  tts = load_tts_model(selected_voice)