hwajjala commited on
Commit
edc9052
β€’
1 Parent(s): c894abc

Fix description

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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.4, the image can be automatically REJECTED as NOT as base
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.