albin commited on
Commit
df99a21
·
1 Parent(s): aa3bc09
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -127,7 +127,7 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
127
  print("Data types:", dataFrame_features[selected_columns].dtypes)
128
 
129
  prediction_proba = model.predict_proba(data)[:, 1]
130
- threshold = 0.1
131
  output = 1 if prediction_proba >= threshold else 0
132
 
133
  output_text = "Legitimate URL" if output == 1 else "Phishing URL"
 
127
  print("Data types:", dataFrame_features[selected_columns].dtypes)
128
 
129
  prediction_proba = model.predict_proba(data)[:, 1]
130
+ threshold = 0.5
131
  output = 1 if prediction_proba >= threshold else 0
132
 
133
  output_text = "Legitimate URL" if output == 1 else "Phishing URL"