update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import torch
 from ultralyticsplus import YOLO, render_result
+import os
+
+HF_TOKEN = os.getenv('HF_TOKEN')
 
 yolo_model = YOLO('eeshawn11/naruto_hand_seal_detection')
 yolo_model.overrides['max_det'] = 10
@@ -28,9 +31,9 @@ def clear():
     """
     Reset inputs.
     """
-    image_upload = gr.update(value=None)
-    conf_slider = gr.update(value=0.5)
-    return image_upload, conf_slider
+    return gr.update(value=None), gr.update(value=0.5)
+
+callback = gr.HuggingFaceDatasetSaver(hf_token=HF_TOKEN, dataset_name="crowdsourced_hand_seals", private=True)
 
 with gr.Blocks() as demo:
     gr.Markdown("# Naruto Hand Seal Detection with YOLOv8")
@@ -43,25 +46,25 @@ with gr.Blocks() as demo:
                 
                 Hand seals are an integral part of the Naruto universe, used by characters to activate powerful techniques. There are twelve basic seals, each named after an animal in the Chinese Zodiac, and different sequences of hand seals are required for different techniques.
     
-                As a fan of the series, I knew that accurately detecting and classifying hand seals would be a difficult but rewarding challenge, and I was excited to tackle it using my expertise in machine learning and computer vision. One key challenge to overcome would be the lack of a good dataset of labelled images for training, so I had to develop my own. Besides capturing images of myself performing the seals, I augmented my dataset with a YouTube screenshots consisting of both real persons and anime characters performing the seals.
+                As a fan of the series, I knew that accurately detecting and classifying hand seals would be a difficult but rewarding challenge, and I was excited to tackle it using my expertise in machine learning and computer vision. One key challenge to overcome would be the lack of a good dataset of labelled images for training, so I had to develop my own. Besides capturing images of myself performing the seals, I augmented my dataset with YouTube screenshots consisting of both real persons and anime characters performing the seals.
     
                 ### Problem Statement
     
                 The challenge was to develop a model that could accurately identify the hand seal being performed.
                 
-                In this project, I leveraged transfer learning from the <a href="https://github.com/ultralytics/ultralytics" target="_blank">YOLOv8</a> model to customize an object detection model specifically for the hand seals.
+                In this project, I leveraged transfer learning from the <a href="https://github.com/ultralytics/ultralytics" target="_blank">YOLOv8</a> model to customize an object detection model specifically for the hand seals. Developed by the Ultralytics team, YOLOv8 is the latest addition to the YOLO family and offers high performance while being easy to train and use.
             """
             )
     with gr.Row():
         with gr.Column():
-            inputs = [
-                    gr.Image(source="upload", type="pil", label="Image Upload", interactive=True),
-                    gr.Slider(minimum=0.05, maximum=1.0, value=0.5, step=0.05, label="Confidence Threshold"),
-                ]
+            img_input = gr.Image(source="upload", type="pil", label="Image Upload", interactive=True),
+            conf_input = gr.Slider(minimum=0.05, maximum=1.0, value=0.5, step=0.05, label="Confidence Threshold"),
             with gr.Row():
                 clear_form = gr.Button("Reset")
                 submit = gr.Button("Predict")
-        outputs = gr.Image(type="filepath", label="Output Image", interactive=False)
+        with gr.Column():
+            outputs = gr.Image(type="filepath", label="Output Image", interactive=False)
+            flag = gr.Button("Flag")
 
     gr.Markdown(
         """
@@ -70,8 +73,12 @@ with gr.Blocks() as demo:
         """
         )
     
-    clear_form.click(fn=clear, inputs=None, outputs=inputs, show_progress=False)
-    submit.click(fn=seal_detection, inputs=inputs, outputs=outputs)
+    callback.setup([img_input, conf_input, outputs], "flagged_data")
+
+    clear_form.click(fn=clear, inputs=None, outputs=[img_input, conf_input], show_progress=False)
+    submit.click(fn=seal_detection, inputs=[img_input, conf_input], outputs=outputs)
+    flag.click(lambda *args: callback.flag(args), [img_input, conf_input, outputs], None, preprocess=False)
 
-demo.queue(api_open=False, max_size=10)
-demo.launch()
+if __name__ == "__main__":
+    demo.queue(api_open=False, max_size=10)
+    demo.launch()