Update app.py

app.py

@@ -14,7 +14,7 @@ st.set_page_config(
 class TTS:
     def __init__(self, model_variant):
         global device
-        device = torch.device('cuda:0') if torch.cuda.is_available() else 'cpu'
+        device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         torch.cuda.manual_seed(1234) if torch.cuda.is_available() else None
         self.flowgenerator = Glow_model(n_vocab = 70, h_c= 192, f_c = 768, f_c_dp = 256, out_c = 80, k_s = 3, k_s_dec = 5, heads=2, layers_enc = 6).to(device)
         self.voicegenerator = GAN_model().to(device)