update app.py

app.py

@@ -35,7 +35,7 @@ def clear():
     conf_update = gr.Slider.update(value=0.5)
     return image_update, conf_update, image_update
 
-callback = gr.HuggingFaceDatasetSaver(hf_token=HF_TOKEN, dataset_name="crowdsourced_hand_seals", private=True)
+callback = gr.HuggingFaceDatasetJSONSaver(hf_token=HF_TOKEN, dataset_name="crowdsourced_hand_seals", private=True)
 
 with gr.Blocks() as demo:
     gr.Markdown("# Naruto Hand Seal Detection with YOLOv8")
@@ -77,11 +77,11 @@ with gr.Blocks() as demo:
         """
         )
     
-    callback.setup([image, slider, outputs], "flagged_data")
+    callback.setup([image, slider, outputs], "crowdsourced_hand_seals")
 
     clear_form.click(fn=clear, inputs=None, outputs=[image[0], slider[0], outputs], show_progress=False)
     submit.click(fn=seal_detection, inputs=[image[0], slider[0]], outputs=outputs)
-    flag.click(lambda *args: callback.flag(args), inputs=[image[0], slider[0], outputs], outputs=None, preprocess=False, show_progress=False)
+    flag.click(lambda *args: callback.flag([image[0], slider[0], outputs]), inputs=[image[0], slider[0], outputs], outputs=None, preprocess=False)
 
 if __name__ == "__main__":
     demo.queue(api_open=False, max_size=10)