Update app.py

app.py

@@ -90,7 +90,7 @@ def get_inference(text, n_timesteps=20, mel_temp = 0.667, length_scale=0.8, spk=
     return synthesise_mel(text, text_lengths, n_timesteps, mel_temp, length_scale, spk)
 
 
-device = torch.device("cuda")
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model_path = './checkpoints/checkpoint.ckpt'
 vocoder_path = './checkpoints/generator'
 model = load_matcha(model_path, device)