add model

app.py

@@ -49,11 +49,28 @@ tts_model_choice = DEFAULT_TTS_MODEL
 vocoder = load_vocoder()
 
 
-def load_f5tts(ckpt_path=str(cached_path("hf://SWivid/F5-TTS/F5TTS_Base/model_1200000.safetensors"))):
+import os
+from huggingface_hub import cached_path, HfApi
+
+
+
+def load_f5tts():
+    # Carrega o caminho do modelo e o token das variáveis de ambiente
+    ckpt_path = os.getenv("MODEL_CKPT_PATH", "hf://SWivid/F5-TTS/F5TTS_Base/model_1200000.safetensors")
+    token = os.getenv("HUGGINGFACE_TOKEN")
+
+    # Valida se o token está presente
+    if not token:
+        raise ValueError("A variável de ambiente 'HUGGINGFACE_TOKEN' não foi definida.")
+
+    # Carrega o modelo
+    ckpt_path = cached_path(ckpt_path, use_auth_token=token)
+    
     F5TTS_model_cfg = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
     return load_model(DiT, F5TTS_model_cfg, ckpt_path)
 
 
+
 def load_e2tts(ckpt_path=str(cached_path("hf://SWivid/E2-TTS/E2TTS_Base/model_1200000.safetensors"))):
     E2TTS_model_cfg = dict(dim=1024, depth=24, heads=16, ff_mult=4)
     return load_model(UNetT, E2TTS_model_cfg, ckpt_path)