Spaces:
Sleeping
Sleeping
albin
commited on
Commit
·
aa3bc09
1
Parent(s):
ed658f7
debuggung
Browse files
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))
|