cocktailpeanut commited on
Commit
54cbe20
·
1 Parent(s): b58be17
Files changed (1) hide show
  1. app2.py +3 -3
app2.py CHANGED
@@ -132,10 +132,10 @@ def infer(ref_audio_orig, ref_text, gen_text, exp_name, remove_silence):
132
  # Select model
133
  if exp_name == "F5-TTS":
134
  ema_model = F5TTS_ema_model
135
- base_model = F5TTS_base_model
136
  elif exp_name == "E2-TTS":
137
  ema_model = E2TTS_ema_model
138
- base_model = E2TTS_base_model
139
 
140
  # Transcribe reference audio if needed
141
  if not ref_text.strip():
@@ -196,7 +196,7 @@ def infer(ref_audio_orig, ref_text, gen_text, exp_name, remove_silence):
196
  # Inference
197
  gr.Info(f"Generating audio using {exp_name}")
198
  with torch.inference_mode():
199
- generated, _ = base_model.sample(
200
  cond=audio,
201
  text=final_text_list,
202
  duration=duration,
 
132
  # Select model
133
  if exp_name == "F5-TTS":
134
  ema_model = F5TTS_ema_model
135
+ # base_model = F5TTS_base_model
136
  elif exp_name == "E2-TTS":
137
  ema_model = E2TTS_ema_model
138
+ # base_model = E2TTS_base_model
139
 
140
  # Transcribe reference audio if needed
141
  if not ref_text.strip():
 
196
  # Inference
197
  gr.Info(f"Generating audio using {exp_name}")
198
  with torch.inference_mode():
199
+ generated, _ = ema_model.sample(
200
  cond=audio,
201
  text=final_text_list,
202
  duration=duration,