hwajjala commited on
Commit
bc5eea3
β€’
1 Parent(s): 1a4386f

Update hair threshold

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -143,7 +143,7 @@ def predict_fn(input_img):
143
  old_result_probabilty = float(old_lr_probabilities[0][1].round(3))
144
  # get decision string
145
  if result_probabilty > 0.77:
146
- if hair_result_probabilty < 0.5:
147
  logger.info("hair_result_probabilty < 0.5")
148
  result_probabilty = hair_result_probabilty
149
  decision = "AUTO REJECT"
@@ -173,7 +173,7 @@ iface = gr.Interface(
173
  description="""
174
  The model returns the probability of the image being a base body. If
175
  probability > 0.77, the image can be automatically tagged as a base body. If
176
- probability < 0.2, the image can be automatically REJECTED as NOT as base
177
  body. All other cases will be submitted for moderation.
178
 
179
  Please flag if you think the decision is wrong.
 
143
  old_result_probabilty = float(old_lr_probabilities[0][1].round(3))
144
  # get decision string
145
  if result_probabilty > 0.77:
146
+ if hair_result_probabilty < 0.25:
147
  logger.info("hair_result_probabilty < 0.5")
148
  result_probabilty = hair_result_probabilty
149
  decision = "AUTO REJECT"
 
173
  description="""
174
  The model returns the probability of the image being a base body. If
175
  probability > 0.77, the image can be automatically tagged as a base body. If
176
+ probability < 0.25, the image can be automatically REJECTED as NOT as base
177
  body. All other cases will be submitted for moderation.
178
 
179
  Please flag if you think the decision is wrong.