Spaces:
Runtime error
Runtime error
Francisco Cerna Fukuzaki
commited on
Commit
·
bb6c8ae
1
Parent(s):
e40125d
Commit
Browse files
app.py
CHANGED
|
@@ -49,7 +49,7 @@ model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_rel
|
|
| 49 |
|
| 50 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 51 |
|
| 52 |
-
hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags", "AllAideas")
|
| 53 |
|
| 54 |
def yolo(size, iou, conf, im):
|
| 55 |
'''Wrapper fn for gradio'''
|
|
|
|
| 49 |
|
| 50 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 51 |
|
| 52 |
+
hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags", repo_id="AllAideas/demo-iazika-flags")
|
| 53 |
|
| 54 |
def yolo(size, iou, conf, im):
|
| 55 |
'''Wrapper fn for gradio'''
|