Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
a6f6f36
1
Parent(s):
5b935e1
change model path
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ from inference import StyleTTS2
|
|
19 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
20 |
|
21 |
config_path = os.path.join(repo_dir, "Models", "config.yml")
|
22 |
-
models_path = os.path.join(repo_dir, "Models", "
|
23 |
model = StyleTTS2(config_path, models_path).to(device)
|
24 |
|
25 |
# Core inference function
|
|
|
19 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
20 |
|
21 |
config_path = os.path.join(repo_dir, "Models", "config.yml")
|
22 |
+
models_path = os.path.join(repo_dir, "Models", "model.pth")
|
23 |
model = StyleTTS2(config_path, models_path).to(device)
|
24 |
|
25 |
# Core inference function
|