Spaces:
Running
Running
Francisco Cerna Fukuzaki
commited on
Commit
·
bebfd30
1
Parent(s):
b3d85dc
Commit
Browse files- app.py +1 -1
- requirements.txt +1 -0
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",
|
| 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", "AllAideas/demo-iazika-flags")
|
| 53 |
|
| 54 |
def yolo(size, iou, conf, im):
|
| 55 |
'''Wrapper fn for gradio'''
|
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
# pip install -r requirements.txt
|
| 2 |
# base ----------------------------------------
|
|
|
|
| 3 |
gradio>=3.4.1
|
| 4 |
huggingface_hub
|
| 5 |
matplotlib>=3.2.2
|
|
|
|
| 1 |
# pip install -r requirements.txt
|
| 2 |
# base ----------------------------------------
|
| 3 |
+
-f https://download.pytorch.org/whl/cpu/torch_stable.html
|
| 4 |
gradio>=3.4.1
|
| 5 |
huggingface_hub
|
| 6 |
matplotlib>=3.2.2
|