Spaces:

stephenhoang
/

ttsStyleTTS2

Sleeping

stephenhoang commited on 29 days ago

Commit

4ea28e6

1 Parent(s): cd71d82

updata app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ 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", "inference", "model.pth")
 device = "cuda" if torch.cuda.is_available() else "cpu"

 # 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")
 device = "cuda" if torch.cuda.is_available() else "cpu"