dangtr0408 commited on
Commit
5c17ac5
·
1 Parent(s): 2c9c807

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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")]