kmunzwa commited on
Commit
03b270b
·
verified ·
1 Parent(s): 3d40fd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -91,7 +91,7 @@ def classify_image(image):
91
  print(f"Non-Cervix: {prob_non_cervix:.4f} | Cervix: {prob_cervix:.4f}")
92
 
93
  # determine the final prediction label
94
- if prob_cervix > prob_non_cervix:
95
  prediction_text = "Cervix Detected"
96
  else:
97
  prediction_text = "Non-Cervix"
 
91
  print(f"Non-Cervix: {prob_non_cervix:.4f} | Cervix: {prob_cervix:.4f}")
92
 
93
  # determine the final prediction label
94
+ if prob_cervix >= 0.55:
95
  prediction_text = "Cervix Detected"
96
  else:
97
  prediction_text = "Non-Cervix"