Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -98,4 +98,6 @@ def predict_email(email_text):
|
|
98 |
max_label, max_prob = max(results.items(), key=lambda x: x[1])
|
99 |
|
100 |
# Risk levels with higher threshold (85%)
|
101 |
-
|
|
|
|
|
|
98 |
max_label, max_prob = max(results.items(), key=lambda x: x[1])
|
99 |
|
100 |
# Risk levels with higher threshold (85%)
|
101 |
+
if "phishing" in max_label.lower() or "suspicious" in max_label.lower():
|
102 |
+
|
103 |
+
|