Update app.py

app.py

@@ -66,7 +66,8 @@ def parse_envs(folder='./videos'):
 
 def build_interface(iter=3, data_folder='./videos'):
     HF_TOKEN = os.getenv('HF_TOKEN')
-    hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "crowdsourced-robotinder-demo")
+    hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "crowdsourced-robotinder-demo")  # HuggingFace logger instead of local one: https://github.com/gradio-app/gradio/blob/master/gradio/flagging.py
+    # callback = gr.CSVLogger()
     callback = hf_writer
     
     # build gradio interface
@@ -96,7 +97,7 @@ def build_interface(iter=3, data_folder='./videos'):
         btn2 = gr.Button("Next")
 
         # This needs to be called at some point prior to the first call to callback.flag()
-        callback.setup([user_choice, left, right])
+        callback.setup([user_choice, left, right], "flagged_data_points")
         
         btn1.click(fn=replay, inputs=[left, right], outputs=[left, right])
         btn2.click(fn=update, inputs=None, outputs=[left, right])