Update app.py

fix original HuggingFaceDatasetSaver bug

app.py

@@ -4,6 +4,7 @@ import random
 import numpy as np
 
 from datasets import load_dataset
+from .hfserver import HuggingFaceDatasetSaver
 
 # dataset = load_dataset("quantumiracle-git/robotinder-data")
 
@@ -66,7 +67,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")  # HuggingFace logger instead of local one: https://github.com/gradio-app/gradio/blob/master/gradio/flagging.py
+    # 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
+    hf_writer = HuggingFaceDatasetSaver(HF_TOKEN, "crowdsourced-robotinder-demo")
     # callback = gr.CSVLogger()
     callback = hf_writer