Commit

app.py

@@ -48,6 +48,8 @@ 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")
+os.environ["HF_DATASETS_OFFLINE"] = 1
+os.environ["TRANSFORMERS_OFFLINE"] = 1
 
 hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "datasets/AllAideas/demo-iazika-flags")
 
@@ -99,7 +101,6 @@ iface = gr.Interface(yolo,
                         theme="huggingface", 
                         analytics_enabled=False,
                         allow_flagging="manual",
-                        flagging_options=["correcto", "incorrecto", "casi correcto", "error", "otro"],
                         flagging_callback=hf_writer
                         ).launch(enable_queue=True,debug=True)