fix

app.py

@@ -50,27 +50,27 @@ vocoder = load_vocoder()
 
 
 import os
-from huggingface_hub import HfApi
-
-
+from huggingface_hub import hf_hub_download
 
 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")
+    # Carrega o caminho do repositório e o nome do arquivo das variáveis de ambiente
+    repo_id = os.getenv("MODEL_REPO_ID", "SWivid/F5-TTS/F5TTS_Base")
+    filename = os.getenv("MODEL_FILENAME", "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)
+    # Faz o download do modelo do repositório privado
+    ckpt_path = hf_hub_download(repo_id=repo_id, filename=filename, 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)