use proba instead of values for result

main.py

@@ -129,7 +129,6 @@ 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 >= threshold else 0
-    output.astype(int)
     
     output_text = "Legitimate" if output == 1 else "Phishing"