Enable mistake flagging

app.py

@@ -199,7 +199,7 @@ if __name__ == "__main__":
                         show_label=True,
                         value=None,
                     )
-                    # open_domain_flag_btn = gr.Button("Flag Mistake", variant="primary")
+                    open_domain_flag_btn = gr.Button("Flag Mistake", variant="primary")
 
             with gr.Row():
                 gr.Examples(
@@ -210,19 +210,19 @@ if __name__ == "__main__":
                     outputs=[open_domain_output],
                 )
 
-            # open_domain_callback = gr.HuggingFaceDatasetSaver(
-            #     hf_token, "imageomics/bioclip-demo-open-domain-mistakes", private=True
-            # )
-            # open_domain_callback.setup(
-            #     [img_input, rank_dropdown, open_domain_output],
-            #     flagging_dir="logs/flagged",
-            # )
-            # open_domain_flag_btn.click(
-            #     lambda *args: open_domain_callback.flag(args),
-            #     [img_input, rank_dropdown, open_domain_output],
-            #     None,
-            #     preprocess=False,
-            # )
+            open_domain_callback = gr.HuggingFaceDatasetSaver(
+                hf_token, "imageomics/bioclip-demo-open-domain-mistakes", private=True
+            )
+            open_domain_callback.setup(
+                [img_input, rank_dropdown, open_domain_output],
+                flagging_dir="logs/flagged",
+            )
+            open_domain_flag_btn.click(
+                lambda *args: open_domain_callback.flag(args),
+                [img_input, rank_dropdown, open_domain_output],
+                None,
+                preprocess=False,
+            )
 
         with gr.Tab("Zero-Shot"):
             with gr.Row():
@@ -240,7 +240,7 @@ if __name__ == "__main__":
                     zero_shot_output = gr.Label(
                         num_top_classes=k, label="Prediction", show_label=True
                     )
-                    # zero_shot_flag_btn = gr.Button("Flag Mistake", variant="primary")
+                    zero_shot_flag_btn = gr.Button("Flag Mistake", variant="primary")
 
             with gr.Row():
                 gr.Examples(
@@ -251,18 +251,18 @@ if __name__ == "__main__":
                     outputs=[zero_shot_output],
                 )
 
-        # zero_shot_callback = gr.HuggingFaceDatasetSaver(
-        #     hf_token, "imageomics/bioclip-demo-zero-shot-mistakes", private=True
-        # )
-        # zero_shot_callback.setup(
-        #     [img_input, zero_shot_output], flagging_dir="logs/flagged"
-        # )
-        # zero_shot_flag_btn.click(
-        #     lambda *args: zero_shot_callback.flag(args),
-        #     [img_input, zero_shot_output],
-        #     None,
-        #     preprocess=False,
-        # )
+        zero_shot_callback = gr.HuggingFaceDatasetSaver(
+            hf_token, "imageomics/bioclip-demo-zero-shot-mistakes", private=True
+        )
+        zero_shot_callback.setup(
+            [img_input, zero_shot_output], flagging_dir="logs/flagged"
+        )
+        zero_shot_flag_btn.click(
+            lambda *args: zero_shot_callback.flag(args),
+            [img_input, zero_shot_output],
+            None,
+            preprocess=False,
+        )
 
         rank_dropdown.change(
             fn=change_output, inputs=rank_dropdown, outputs=[open_domain_output]