Deploy app (download checkpoint from model repo)

.gitignore

@@ -32,6 +32,3 @@ tensorboard/
 
 
 # Allow checkpoint for HF Space demo
-!Models/
-!Models/Finetune/
-!Models/Finetune/epoch_00000.pth

app.py

@@ -21,7 +21,13 @@ SPEAKER2REFS_PATH = os.path.join(DATA_ROOT, "speaker2refs.json")
 # Repo StyleTTS2-lite-vi (neu app.py nam trong repo thi de "./")
 repo_dir = "./"
 config_path = os.path.join(repo_dir, "Models", "config.yaml")
-models_path = os.path.join(repo_dir, "Models", "Finetune", "current_model.pth")
+# models_path = os.path.join(repo_dir, "Models", "Finetune", "epoch_00000.pth")
+from huggingface_hub import hf_hub_download
+
+CKPT_REPO = "stephenhoang/ttsStyleTTS2-ms152"
+models_path = hf_hub_download(repo_id=CKPT_REPO, filename="epoch_00000.pth")
+config_path = hf_hub_download(repo_id=CKPT_REPO, filename="config.yaml")
+
 
 device = "cuda" if torch.cuda.is_available() else "cpu"