Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -108,19 +108,17 @@ chat_tokenizer_state = None
|
|
| 108 |
|
| 109 |
|
| 110 |
# load models
|
| 111 |
-
F5TTS_model_cfg = dict(
|
| 112 |
-
dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4
|
| 113 |
-
)
|
| 114 |
E2TTS_model_cfg = dict(dim=1024, depth=24, heads=16, ff_mult=4)
|
| 115 |
|
| 116 |
F5TTS_ema_model = load_custom(
|
| 117 |
-
"
|
| 118 |
)
|
| 119 |
E2TTS_ema_model = load_custom(
|
| 120 |
-
"
|
| 121 |
)
|
| 122 |
E2TTS_ema_model2 = load_custom(
|
| 123 |
-
"
|
| 124 |
)
|
| 125 |
|
| 126 |
def chunk_text(text, max_chars=135):
|
|
|
|
| 108 |
|
| 109 |
|
| 110 |
# load models
|
| 111 |
+
F5TTS_model_cfg = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
|
|
|
|
|
|
|
| 112 |
E2TTS_model_cfg = dict(dim=1024, depth=24, heads=16, ff_mult=4)
|
| 113 |
|
| 114 |
F5TTS_ema_model = load_custom(
|
| 115 |
+
"hf://Gregniuki/F5-tts_English_German_Polish/English/model_222600.pt", "", F5TTS_model_cfg
|
| 116 |
)
|
| 117 |
E2TTS_ema_model = load_custom(
|
| 118 |
+
"hf://Gregniuki/F5-tts_English_German_Polish/Polish2/model_1200000.pt", "", F5TTS_model_cfg
|
| 119 |
)
|
| 120 |
E2TTS_ema_model2 = load_custom(
|
| 121 |
+
"hf://Gregniuki/F5-tts_English_German_Polish/Polish/model_500000.pt", "", F5TTS_model_cfg
|
| 122 |
)
|
| 123 |
|
| 124 |
def chunk_text(text, max_chars=135):
|