Update app.py

app.py

@@ -43,7 +43,7 @@ def diarize_audio(audio_path, num_speakers):
 
         # Load audio for pyannote
         waveform, sample_rate = torchaudio.load(wav_path)
-        audio_dict = {"waveform": waveform.to(device), "sample_rate": sample_rate}
+        audio_dict = {"waveform": waveform, "sample_rate": sample_rate}
 
         # Configure pipeline with number of speakers
         pipeline_params = {"num_speakers": num_speakers}