Add feedback writer

app.py

@@ -15,8 +15,14 @@ CLIP_MODEL_NAME = "ViT-B/16"
 TEXT_PROMPTS_FILE_NAME = "text_prompts.json"
 LOGISTIC_REGRESSION_MODEL_FILE_NAME = "logistic_regression_l1_oct_2.pkl"
 
+HF_TOKEN = os.getenv('HF_TOKEN')
+hf_writer = gr.HuggingFaceDatasetSaver(
+    HF_TOKEN, "Roblox/basebody_feedback"
+)
 
-clip_model, preprocess = clip.load(CLIP_MODEL_NAME, device="cpu")
+clip_model, preprocess = clip.load(
+    CLIP_MODEL_NAME, device="cpu"
+)
 
 with open(
     os.path.join(os.path.dirname(__file__), TEXT_PROMPTS_FILE_NAME), "r"
@@ -74,6 +80,7 @@ iface = gr.Interface(
     fn=predict_fn,
     inputs="image",
     outputs="text",
-    allow_flagging="manual"
+    allow_flagging="manual",
+    flagging_callback=hf_writer
 )
 iface.launch()