Move audio back to CPU

app.py

@@ -76,6 +76,8 @@ def process_fn(audio_path, transcribe):#, demo):
     # Resample audio back to the original sampling rate
     audio = torchaudio.functional.resample(audio, 22050, fs)
 
+    audio = audio.cpu()
+
     # Create a temporary directory for output
     os.makedirs('_outputs', exist_ok=True)
     # Create a path for saving the audio