Update app.py

app.py

@@ -5,7 +5,7 @@ import torchaudio
 import os
 
 
-MODEL_DIR = "my_model"
+# MODEL_DIR = "my_model"
 OUTPUT_DIR = "outputs"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
@@ -15,17 +15,10 @@ from huggingface_hub import hf_hub_download
 # ------------------------
 # Download model files from Hugging Face if not present
 # ------------------------
-MODEL_REPO = "MariaKaiser/EGTTS_finetuned"  # replace with your repo name
-
-if not os.path.exists(os.path.join(MODEL_DIR, "config.json")):
-    os.makedirs(MODEL_DIR, exist_ok=True)
-    config_file = hf_hub_download(repo_id=MODEL_REPO, filename="config.json", cache_dir=MODEL_DIR)
-    vocab_file = hf_hub_download(repo_id=MODEL_REPO, filename="vocab.json", cache_dir=MODEL_DIR)
-    model_file = hf_hub_download(repo_id=MODEL_REPO, filename="model.pth", cache_dir=MODEL_DIR)
-else:
-    config_file = os.path.join(MODEL_DIR, "config.json")
-    vocab_file = os.path.join(MODEL_DIR, "vocab.json")
-    model_file = os.path.join(MODEL_DIR, "model.pth")
+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)
 
 from TTS.tts.models.xtts import Xtts
 from TTS.tts.configs.xtts_config import XttsConfig