hwajjala commited on
Commit
509301a
1 Parent(s): bb70cd9

Bugfix decision

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -117,7 +117,8 @@ def predict_fn(input_img):
117
  if hair_result_probabilty < 0.5:
118
  result_probabilty = hair_result_probabilty
119
  decision = "AUTO REJECT"
120
- decision = "AUTO ACCEPT"
 
121
  elif result_probabilty < 0.4:
122
  decision = "AUTO REJECT"
123
  else:
@@ -136,7 +137,7 @@ iface = gr.Interface(
136
  outputs="text",
137
  description="""
138
  The model returns the probability of the image being a base body. If
139
- probability > 0.95, the image can be automatically tagged as a base body. If
140
  probability < 0.4, the image can be automatically REJECTED as NOT as base
141
  body. All other cases will be submitted for moderation.
142
 
 
117
  if hair_result_probabilty < 0.5:
118
  result_probabilty = hair_result_probabilty
119
  decision = "AUTO REJECT"
120
+ else:
121
+ decision = "AUTO ACCEPT"
122
  elif result_probabilty < 0.4:
123
  decision = "AUTO REJECT"
124
  else:
 
137
  outputs="text",
138
  description="""
139
  The model returns the probability of the image being a base body. If
140
+ probability > 0.77, the image can be automatically tagged as a base body. If
141
  probability < 0.4, the image can be automatically REJECTED as NOT as base
142
  body. All other cases will be submitted for moderation.
143