Update main.py

main.py

@@ -126,8 +126,9 @@ async def upload_audio(file: UploadFile = File(...)):
 
     # Perform inference
     print("Performing inference with Wav2Vec2 model...")
-    output = (model, input_values)
-    logits = output.logits
+    
+    logits = model(input_values).logits
+    
     print("Inference complete. Shape of logits:", logits.shape)
 
     # Get predictions