Update app.py

app.py

@@ -24,7 +24,8 @@ def predict_voice(audio_file_path):
         features = custom_feature_extraction(audio_file_path)
         
         with torch.no_grad():
-            outputs = model(inputs=features)
+            # Directly pass the features tensor to the model
+            outputs = model(features)
         
         logits = outputs.logits
         predicted_index = logits.argmax()