albin commited on
Commit
aa3bc09
·
1 Parent(s): ed658f7
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -130,7 +130,7 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
130
  threshold = 0.1
131
  output = 1 if prediction_proba >= threshold else 0
132
 
133
- output_text = "Legitimate" if output == 1 else "Phishing"
134
 
135
  # Render index.html with prediction results
136
  json_compatible_resp_data = jsonable_encoder(Resp(url=requess.url, label=output_text))
 
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"
134
 
135
  # Render index.html with prediction results
136
  json_compatible_resp_data = jsonable_encoder(Resp(url=requess.url, label=output_text))