Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
5c17ac5
1
Parent(s):
2c9c807
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ from inference import StyleTTS2
|
|
17 |
|
18 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
19 |
config_path = os.path.join(repo_dir, "Models", "config.yaml")
|
20 |
-
models_path = os.path.join(repo_dir, "Models", "model.pth")
|
21 |
model = StyleTTS2(config_path, models_path).eval().to(device)
|
22 |
voice_path = os.path.join(repo_dir, "reference_audio")
|
23 |
eg_voices = [os.path.join(voice_path,"vn_1.wav"), os.path.join(voice_path,"vn_2.wav")]
|
|
|
17 |
|
18 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
19 |
config_path = os.path.join(repo_dir, "Models", "config.yaml")
|
20 |
+
models_path = os.path.join(repo_dir, "Models", "inference", "model.pth")
|
21 |
model = StyleTTS2(config_path, models_path).eval().to(device)
|
22 |
voice_path = os.path.join(repo_dir, "reference_audio")
|
23 |
eg_voices = [os.path.join(voice_path,"vn_1.wav"), os.path.join(voice_path,"vn_2.wav")]
|