use proba instead of values for result

main.py

@@ -128,7 +128,7 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
 
     prediction_proba = model.predict_proba(data)[:, 1]
     threshold = 0.1
-    output = 1 if prediction_proba[0][1] >= threshold else 0
+    output = 1 if prediction_proba >= threshold else 0
     output.astype(int)
     
     output_text = "Legitimate" if output == 1 else "Phishing"