Update app.py
Browse files
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()
|