Commit

app.py

@@ -50,7 +50,7 @@ model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_rel
 #model = torch.hub.load('path/to/yolov5', 'custom', path='/content/yolov56.pt', source='local')  # local repo
 
 HF_TOKEN = os.getenv("HF_TOKEN")
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags")
+#hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags")
 
 def yolo(size, iou, conf, im):
     '''Wrapper fn for gradio'''
@@ -98,10 +98,7 @@ iface = gr.Interface(yolo,
                         article=article, 
                         examples=examples,
                         theme="huggingface", 
-                        analytics_enabled=False,
-                        allow_flagging="manual",
-                        flagging_options=["correcto", "incorrecto", "casi correcto", "error", "otro"],
-                        flagging_callback=hf_writer).launch(
+                        analytics_enabled=False).launch(
     debug=True)
 
 iface.launch()