Update app.py
Browse files
app.py
CHANGED
|
@@ -89,7 +89,7 @@ def get_inference(text, n_timesteps=20, mel_temp = 0.667, length_scale=0.8, spk=
|
|
| 89 |
return synthesise_mel(text, text_lengths, n_timesteps, mel_temp, length_scale, spk)
|
| 90 |
|
| 91 |
|
| 92 |
-
device = torch.device("
|
| 93 |
model_path = './checkpoints/checkpoint.ckpt'
|
| 94 |
vocoder_path = './checkpoints/generator'
|
| 95 |
model = load_matcha(model_path, device)
|
|
|
|
| 89 |
return synthesise_mel(text, text_lengths, n_timesteps, mel_temp, length_scale, spk)
|
| 90 |
|
| 91 |
|
| 92 |
+
device = torch.device("cuda")
|
| 93 |
model_path = './checkpoints/checkpoint.ckpt'
|
| 94 |
vocoder_path = './checkpoints/generator'
|
| 95 |
model = load_matcha(model_path, device)
|