Update app.py

app.py

@@ -112,7 +112,7 @@ def predict(audio_record, audio_upload, threshold):
 
         # Feature extraction
         inputs = mel_spectrogram(chunk)
-        inputs = log_mel_spectrogram(inputs).to(device).unsqueeze(0)
+        inputs = log_mel_spectrogram(inputs).unsqueeze(0)
 
         # Model inference
         with torch.no_grad():