Update app.py

app.py

@@ -39,7 +39,7 @@ def save_audio(samples: torch.Tensor):
     save_path = "/audio_output"
     assert samples.dim() == 2 or samples.dim() == 3
 
-    samples = samples.detach().gpu()
+    samples = samples.detach().cpu()
     if samples.dim() == 2:
         samples = samples[None, ...]