Update app.py

app.py

@@ -12,6 +12,22 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+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")
+
 # Load model
 config = XttsConfig()
 config.load_json(os.path.join(MODEL_DIR, "config.json"))