use proba instead of values for result

main.py

@@ -126,10 +126,11 @@ 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_proba = model.predict_proba(data)
+    prediction_proba = model.predict_proba(data)[:, 1]
     threshold = 0.1
     output = 1 if prediction_proba[0][1] >= threshold else 0
-
+    output.astype(int)
+    
     output_text = "Legitimate" if output == 1 else "Phishing"
 
     # Render index.html with prediction results