Spaces:
Runtime error
Runtime error
Commit
·
54cbe20
1
Parent(s):
b58be17
update
Browse files
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, _ =
|
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,
|