hwajjala commited on
Commit
a59bf63
β€’
1 Parent(s): c86df50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -23,11 +23,6 @@ LOGISTIC_REGRESSION_OLD_MODEL_FILE_NAME = "logistic_regression_l1_oct_2.pkl"
23
  LOGISTIC_REGRESSION_MODEL_FILE_NAME = "logistic_regression_l1_oct_26.pkl"
24
  HAIR_RF_CLASSIFIER_MODEL_FILE_NAME = "hairclassifier_rf.pkl"
25
 
26
- HF_TOKEN = os.getenv('HF_TOKEN')
27
- hf_writer = gr.HuggingFaceDatasetSaver(
28
- HF_TOKEN, "Roblox/basebody_feedback"
29
- )
30
-
31
  clip_model, preprocess = clip.load(
32
  CLIP_MODEL_NAME, device="cpu"
33
  )
@@ -184,7 +179,6 @@ iface = gr.Interface(
184
  ": decision should be accept",
185
  ": decision should be reject",
186
  ": decision should be moderation"
187
- ],
188
- flagging_callback=hf_writer
189
  )
190
  iface.launch()
 
23
  LOGISTIC_REGRESSION_MODEL_FILE_NAME = "logistic_regression_l1_oct_26.pkl"
24
  HAIR_RF_CLASSIFIER_MODEL_FILE_NAME = "hairclassifier_rf.pkl"
25
 
 
 
 
 
 
26
  clip_model, preprocess = clip.load(
27
  CLIP_MODEL_NAME, device="cpu"
28
  )
 
179
  ": decision should be accept",
180
  ": decision should be reject",
181
  ": decision should be moderation"
182
+ ]
 
183
  )
184
  iface.launch()