Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -110,7 +110,7 @@ def generate_ocr(method, image):
|
|
110 |
spam_prob = probs[0][1].item() # Probability of Spam
|
111 |
|
112 |
# Adjust classification based on threshold
|
113 |
-
label = "Spam" if spam_prob > 0.
|
114 |
|
115 |
# Save results using external function
|
116 |
save_results_to_repo(text_output, label)
|
|
|
110 |
spam_prob = probs[0][1].item() # Probability of Spam
|
111 |
|
112 |
# Adjust classification based on threshold
|
113 |
+
label = "Spam" if spam_prob > 0.8 else "Not Spam"
|
114 |
|
115 |
# Save results using external function
|
116 |
save_results_to_repo(text_output, label)
|