Fix description
Browse files
app.py
CHANGED
@@ -77,6 +77,7 @@ with open(
|
|
77 |
|
78 |
logger.info("Logistic regression model loaded, coefficients: ")
|
79 |
|
|
|
80 |
def get_text_features(text_prompts):
|
81 |
all_text_features = []
|
82 |
with torch.no_grad():
|
@@ -89,6 +90,7 @@ def get_text_features(text_prompts):
|
|
89 |
all_text_features = all_text_features.cpu()
|
90 |
return all_text_features
|
91 |
|
|
|
92 |
all_text_features = get_text_features(text_prompts)
|
93 |
hair_text_features = get_text_features(hair_text_prompts)
|
94 |
old_text_features = get_text_features(text_prompts_old)
|
@@ -166,7 +168,7 @@ iface = gr.Interface(
|
|
166 |
description="""
|
167 |
The model returns the probability of the image being a base body. If
|
168 |
probability > 0.77, the image can be automatically tagged as a base body. If
|
169 |
-
probability < 0.
|
170 |
body. All other cases will be submitted for moderation.
|
171 |
|
172 |
Please flag if you think the decision is wrong.
|
|
|
77 |
|
78 |
logger.info("Logistic regression model loaded, coefficients: ")
|
79 |
|
80 |
+
|
81 |
def get_text_features(text_prompts):
|
82 |
all_text_features = []
|
83 |
with torch.no_grad():
|
|
|
90 |
all_text_features = all_text_features.cpu()
|
91 |
return all_text_features
|
92 |
|
93 |
+
|
94 |
all_text_features = get_text_features(text_prompts)
|
95 |
hair_text_features = get_text_features(hair_text_prompts)
|
96 |
old_text_features = get_text_features(text_prompts_old)
|
|
|
168 |
description="""
|
169 |
The model returns the probability of the image being a base body. If
|
170 |
probability > 0.77, the image can be automatically tagged as a base body. If
|
171 |
+
probability < 0.2, the image can be automatically REJECTED as NOT as base
|
172 |
body. All other cases will be submitted for moderation.
|
173 |
|
174 |
Please flag if you think the decision is wrong.
|