Update app.py

app.py

@@ -2,7 +2,11 @@
 from PIL import Image
 import gradio as gr
 from ultralytics import YOLO
+import os
 
+## Where to store flagged data
+HF_TOKEN = os.getenv('HF_TOKEN')
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "Crowd-sourced images")
 
 # Load your trained model
 model = YOLO('best.pt')
@@ -52,6 +56,9 @@ platform = gr.Interface( fn = predict,
                         description="""
                         Introducing a revolutionary computer-aided detection tool designed to enhance the efficiency of clinicians in detecting pneumothorax in chest X-ray images. 
                         """,
+                        allow_flagging="manual",
+                        flagging_options=["Wrong detection", "Ambiguous", "Other"],
+                        flagging_callback=hf_writer,
                         article = """ 
                         It is crucial to emphasize that while this tool serves as a valuable research aid, 
                         it is not intended to replace clinical guidelines,