Update app.py

app.py

@@ -124,9 +124,12 @@ def decode_audio(encoded_file_path):
         with torch.no_grad():
             waveform = semanticodec.decode(tokens)
 
+        # Move waveform to CPU for saving
+        waveform_cpu = waveform.cpu()
+
         # Save to a temporary WAV file
         temp_wav_path = tempfile.mktemp(suffix=".wav")
-        torchaudio.save(temp_wav_path, waveform.squeeze(0).cpu(), sample_rate)
+        torchaudio.save(temp_wav_path, waveform_cpu.squeeze(0), sample_rate)
         return temp_wav_path
 
     except Exception as e: