winamnd commited on
Commit
39fa45a
·
verified ·
1 Parent(s): cfa8cd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.8 else "Not Spam"
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.6 else "Not Spam"
114
 
115
  # Save results using external function
116
  save_results_to_repo(text_output, label)