use proba instead of values for result

main.py

@@ -126,8 +126,9 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
     print("Data sent to predict:", data)
     print("Data types:", dataFrame_features[selected_columns].dtypes)
 
-    prediction = model.predict(data)
-    output = prediction[0]
+    prediction_proba = model.predict_proba(data)
+    threshold = 0.1
+    output = 1 if prediction_proba[0][1] >= threshold else 0
 
     output_text = "Legitimate" if output == 1 else "Phishing"