Gregniuki commited on
Commit
dd00517
·
verified ·
1 Parent(s): 10fac73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -101,7 +101,7 @@ E2TTS_ema_model = load_model(
101
  E2TTS_ema_model2 = load_model(
102
  "Gregniuki", "F5-tts_English_German_Polish", "Polish", DiT, F5TTS_model_cfg, 500000
103
  )
104
- E2TTS_ema_model3 = load_model(
105
  "SWivid", "F5-TTS", "F5TTS_Base", DiT, F5TTS_model_cfg, 1200000
106
  )
107
  E2TTS_ema_model4 = load_model(
@@ -144,8 +144,8 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence,
144
  ema_model = E2TTS_ema_model
145
  elif exp_name == "Deutsh":
146
  ema_model = E2TTS_ema_model2
147
- # elif exp_name == "F5TTS":
148
- # ema_model = E2TTS_ema_model3
149
  elif exp_name == "E2TTS":
150
  ema_model = E2TTS_ema_model4
151
 
 
101
  E2TTS_ema_model2 = load_model(
102
  "Gregniuki", "F5-tts_English_German_Polish", "Polish", DiT, F5TTS_model_cfg, 500000
103
  )
104
+ E2TTS_ema_model3 = load_model(
105
  "SWivid", "F5-TTS", "F5TTS_Base", DiT, F5TTS_model_cfg, 1200000
106
  )
107
  E2TTS_ema_model4 = load_model(
 
144
  ema_model = E2TTS_ema_model
145
  elif exp_name == "Deutsh":
146
  ema_model = E2TTS_ema_model2
147
+ elif exp_name == "F5TTS":
148
+ ema_model = E2TTS_ema_model3
149
  elif exp_name == "E2TTS":
150
  ema_model = E2TTS_ema_model4
151