Spaces:

AllAideas
/

demoIAZIKA

Running

fcernafukuzaki commited on Oct 23, 2023

Commit

d4aa952

1 Parent(s): 0cfc9ed

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,8 +38,8 @@ torch.hub.download_url_to_file('https://i.pinimg.com/originals/c2/ce/e0/c2cee056
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, autoshape=True, trust_repo=True)
-# HF_TOKEN = os.getenv("ZIKA_TOKEN_WRITE")
-# hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags")
 def getQuantity(string):
     contador_raw = ''.join(string.split(" ")[3:])

 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, autoshape=True, trust_repo=True)
+HF_TOKEN = os.getenv("ZIKA_TOKEN_WRITE")
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags")
 def getQuantity(string):
     contador_raw = ''.join(string.split(" ")[3:])