Spaces:
Running
Running
Francisco Cerna
commited on
Commit
·
e00d556
1
Parent(s):
11513d6
Update app.py
Browse files
app.py
CHANGED
@@ -47,12 +47,13 @@ torch.hub.download_url_to_file('https://i.pinimg.com/originals/c2/ce/e0/c2cee056
|
|
47 |
#model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # force_reload=True to update
|
48 |
|
49 |
#model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt') # local model o google colab
|
50 |
-
model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, autoshape=True) # local model o google colab
|
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"
|
56 |
#hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags", "AllAideas", False)
|
57 |
|
58 |
def getQuantity(string):
|
|
|
47 |
#model = torch.hub.load('ultralytics/yolov5', 'yolov5s') # force_reload=True to update
|
48 |
|
49 |
#model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt') # local model o google colab
|
50 |
+
#model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, autoshape=True) # local model o google colab
|
51 |
#model = torch.hub.load('path/to/yolov5', 'custom', path='/content/yolov56.pt', source='local') # local repo
|
52 |
+
model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, autoshape=True, trust_repo=True)
|
53 |
|
54 |
HF_TOKEN = os.getenv("ZIKA_TOKEN_WRITE")
|
55 |
#hf_writer = gr.HuggingFaceDatasetSaver(hf_foken=HF_TOKEN, dataset_name="AllAideas/demo-iazika-flags")
|
56 |
+
hf_writer = gr.HuggingFaceDatasetSaver(hf_foken=HF_TOKEN, dataset_name="AllAideas/demo-iazika-flags")
|
57 |
#hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags", "AllAideas", False)
|
58 |
|
59 |
def getQuantity(string):
|