Spaces:
Running
Running
Francisco Cerna Fukuzaki
commited on
Commit
·
d7e5b97
1
Parent(s):
bebfd30
Commit
Browse files- app.py +1 -1
- requirements.txt +0 -3
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/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", "datasets/AllAideas/demo-iazika-flags")
|
| 53 |
|
| 54 |
def yolo(size, iou, conf, im):
|
| 55 |
'''Wrapper fn for gradio'''
|
requirements.txt
CHANGED
|
@@ -1,8 +1,5 @@
|
|
| 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
|
| 7 |
numpy>=1.18.5
|
| 8 |
opencv-python-headless
|
|
|
|
| 1 |
# pip install -r requirements.txt
|
| 2 |
# base ----------------------------------------
|
|
|
|
|
|
|
|
|
|
| 3 |
matplotlib>=3.2.2
|
| 4 |
numpy>=1.18.5
|
| 5 |
opencv-python-headless
|