Spaces:
Running
Running
Francisco Cerna
commited on
Commit
·
11513d6
1
Parent(s):
0fada95
Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,8 @@ model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_rel
|
|
51 |
#model = torch.hub.load('path/to/yolov5', 'custom', path='/content/yolov56.pt', source='local') # local repo
|
52 |
|
53 |
HF_TOKEN = os.getenv("ZIKA_TOKEN_WRITE")
|
54 |
-
hf_writer = gr.HuggingFaceDatasetSaver(hf_foken=HF_TOKEN, dataset_name="AllAideas/demo-iazika-flags")
|
|
|
55 |
#hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags", "AllAideas", False)
|
56 |
|
57 |
def getQuantity(string):
|
@@ -147,7 +148,7 @@ iface = gr.Interface(yolo,
|
|
147 |
flagging_options=["Correcto", "Incorrecto", "Casi correcto", "Error", "Otro"],
|
148 |
flagging_callback=hf_writer)
|
149 |
|
150 |
-
iface.launch(enable_queue=True, debug=True)
|
151 |
|
152 |
"""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).
|
153 |
|
|
|
51 |
#model = torch.hub.load('path/to/yolov5', 'custom', path='/content/yolov56.pt', source='local') # local repo
|
52 |
|
53 |
HF_TOKEN = os.getenv("ZIKA_TOKEN_WRITE")
|
54 |
+
#hf_writer = gr.HuggingFaceDatasetSaver(hf_foken=HF_TOKEN, dataset_name="AllAideas/demo-iazika-flags")
|
55 |
+
hf_writer = gr.HuggingFaceDatasetSaver(hf_foken=HF_TOKEN, dataset_name="AllAideas/demo-iazika-flags", trust_repo=True)
|
56 |
#hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags", "AllAideas", False)
|
57 |
|
58 |
def getQuantity(string):
|
|
|
148 |
flagging_options=["Correcto", "Incorrecto", "Casi correcto", "Error", "Otro"],
|
149 |
flagging_callback=hf_writer)
|
150 |
|
151 |
+
iface.launch(ache_examples=True, enable_queue=True, debug=True)
|
152 |
|
153 |
"""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).
|
154 |
|