Update app.py

app.py

@@ -38,8 +38,8 @@ filename = 'test' #@param {type: "string"}
 audio_path = f'/content/VITS-Aatrox/{filename}.wav'
 stn_tst = get_text(text, hps)
 with torch.no_grad():
-    x_tst = stn_tst.cuda().unsqueeze(0)
-    x_tst_lengths = torch.LongTensor([stn_tst.size(0)]).cuda()
+    x_tst = stn_tst.unsqueeze(0)
+    x_tst_lengths = torch.LongTensor([stn_tst.size(0)])
     audio = net_g.infer(x_tst, x_tst_lengths, noise_scale=.667, noise_scale_w=0.8, length_scale=length_scale)[0][0,0].data.cpu().float().numpy()
 ipd.display(ipd.Audio(audio, rate=hps.data.sampling_rate))