cesar commited on
Commit
0398a37
1 Parent(s): 7709ecd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -38,8 +38,8 @@ torch.hub.download_url_to_file('https://i.pinimg.com/originals/c2/ce/e0/c2cee056
38
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, autoshape=True, trust_repo=True)
39
 
40
 
41
- HF_TOKEN = os.getenv("ZIKA_TOKEN_WRITE")
42
- hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags")
43
 
44
  def getQuantity(string):
45
  contador_raw = ''.join(string.split(" ")[3:])
@@ -135,7 +135,7 @@ iface = gr.Interface(yolo,
135
  #flagging_callback=hf_writer
136
  )
137
 
138
- iface.launch(enable_queue=True, debug=True, share=True)
139
 
140
  """For YOLOv5 PyTorch Hub inference with **PIL**, **OpenCV**, **Numpy** or **PyTorch** inputs please see the full [YOLOv5 PyTorch Hub Tutorial](https://github.com/ultralytics/yolov5/issues/36).
141
  ## Citation
 
38
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, autoshape=True, trust_repo=True)
39
 
40
 
41
+ #HF_TOKEN = os.getenv("ZIKA_TOKEN_WRITE")
42
+ #hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags")
43
 
44
  def getQuantity(string):
45
  contador_raw = ''.join(string.split(" ")[3:])
 
135
  #flagging_callback=hf_writer
136
  )
137
 
138
+ iface.launch(enable_queue=True, debug=True)
139
 
140
  """For YOLOv5 PyTorch Hub inference with **PIL**, **OpenCV**, **Numpy** or **PyTorch** inputs please see the full [YOLOv5 PyTorch Hub Tutorial](https://github.com/ultralytics/yolov5/issues/36).
141
  ## Citation