Disable Flag Mistake buttons for Open-Domain and Zero-Shot predictions

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]