Update app.py

app.py

@@ -84,13 +84,9 @@ if audio_file:
     # ✅ Preprocess Audio with Processor (Corrected)
     # ================================
     inputs = processor(adversarial_waveform.squeeze().numpy(), sampling_rate=16000, return_tensors="pt", padding=True)
-    
-    # Ensure the input has batch dimension (even if it's one example)
-    if len(inputs.input_features.shape) == 1:
-        inputs.input_features = inputs.input_features.unsqueeze(0)
 
-    # Move the input features to the correct device (GPU/CPU)
-    input_features = inputs.input_features.to("cuda" if torch.cuda.is_available() else "cpu")
+    # Ensure the input has batch dimension (even if it's one example)
+    input_features = inputs["input_features"].to("cuda" if torch.cuda.is_available() else "cpu")
 
     # ================================
     # ✅ Fast Transcription Processing with Conformer