Francisco Cerna Fukuzaki commited on
Commit
22ba749
1 Parent(s): ca2b7ea
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,8 +48,8 @@ model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_rel
48
  #model = torch.hub.load('path/to/yolov5', 'custom', path='/content/yolov56.pt', source='local') # local repo
49
 
50
  HF_TOKEN = os.getenv("HF_TOKEN")
51
- os.environ["HF_DATASETS_OFFLINE"] = 1
52
- os.environ["TRANSFORMERS_OFFLINE"] = 1
53
 
54
  hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "datasets/AllAideas/demo-iazika-flags")
55
 
 
48
  #model = torch.hub.load('path/to/yolov5', 'custom', path='/content/yolov56.pt', source='local') # local repo
49
 
50
  HF_TOKEN = os.getenv("HF_TOKEN")
51
+ os.environ["HF_DATASETS_OFFLINE"] = "1"
52
+ os.environ["TRANSFORMERS_OFFLINE"] = "1"
53
 
54
  hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "datasets/AllAideas/demo-iazika-flags")
55