Update hair threshold
Browse files
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.
|
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.
|
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.24:
|
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.24, 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.
|