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