ElMETRICO commited on
Commit
c90ccdd
·
verified ·
1 Parent(s): 220f24a

Loosen species safety gate for diseased pet closeup demo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -152,7 +152,7 @@ def predict_species_open_set_strict(image_path):
152
  final_output = "uncertain_possible_unknown"
153
  elif top1_prob < 0.90:
154
  final_output = "unknown_or_unclear_image"
155
- elif margin < 0.30:
156
  final_output = "uncertain_species"
157
  else:
158
  final_output = top1_class
 
152
  final_output = "uncertain_possible_unknown"
153
  elif top1_prob < 0.90:
154
  final_output = "unknown_or_unclear_image"
155
+ elif margin < 0.05:
156
  final_output = "uncertain_species"
157
  else:
158
  final_output = top1_class