Update app.py

app.py

@@ -15,24 +15,38 @@ from huggingface_hub import hf_hub_download
 # ------------------------
 # Download model files from Hugging Face if not present
 # ------------------------
-MODEL_DIR = "/tmp/my_model"
-os.makedirs(MODEL_DIR, exist_ok=True)
-for fname in ["config.json", "vocab.json", "model.pth"]:
-    hf_hub_download("MariaKaiser/EGTTS_finetuned", filename=fname, cache_dir=MODEL_DIR)
+MODEL_DIR = "my_model"
+config_path = hf_hub_download(
+    repo_id="MariaKaiser/EGTTS_finetuned",
+    filename="config.json",
+    cache_dir=MODEL_DIR
+)
+
+vocab_path = hf_hub_download(
+    repo_id="MariaKaiser/EGTTS_finetuned",
+    filename="vocab.json",
+    cache_dir=MODEL_DIR
+)
+
+model_path = hf_hub_download(
+    repo_id="MariaKaiser/EGTTS_finetuned",
+    filename="model.pth",
+    cache_dir=MODEL_DIR
+)
 
 from TTS.tts.models.xtts import Xtts
 from TTS.tts.configs.xtts_config import XttsConfig
 
 # Load model
 config = XttsConfig()
-config.load_json(os.path.join(MODEL_DIR, "config.json"))
+config.load_json(config_path)
 
 model = Xtts.init_from_config(config)
 model.load_checkpoint(
     config,
-    checkpoint_dir=MODEL_DIR,
+    checkpoint_dir= os.path.dirname(model_path),
     use_deepspeed=False,
-    vocab_path=os.path.join(MODEL_DIR, "vocab.json")
+    vocab_path= vocab_path
 )
 model.to(device)