Update app/app_savta.py

app/app_savta.py

@@ -5,6 +5,12 @@ import os
 from fastai.vision.all import *
 import gradio as gr
 
+############### HF ###########################
+
+HF_TOKEN = os.getenv('HF_TOKEN')
+
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "savtadepth-flags")
+
 ############## DVC ################################
 
 PROD_MODEL_PATH = "src/models"
@@ -73,7 +79,7 @@ def sepia(input_img):
 
 
 def main():
-	iface = gr.Interface(sepia, gr.inputs.Image(shape=(640,480),type='numpy'), "image", title = title, flagging_options=["incorrect", "worst","ambiguous"], allow_flagging = "manual", description = description, article = article, examples = examples, theme ="peach",thumbnail=thumbnail)
+	iface = gr.Interface(sepia, gr.inputs.Image(shape=(640,480),type='numpy'), "image", title = title, flagging_options=["incorrect", "worst","ambiguous"], allow_flagging = "manual",flagging_callback=hf_writer, description = description, article = article, examples = examples, theme ="peach",thumbnail=thumbnail)
 
 	iface.launch(enable_queue=True)
 # enable_queue=True,auth=("admin", "pass1234")