Commit

app.py

@@ -36,8 +36,7 @@ import gradio as gr
 import torch
 from PIL import Image
 
-HF_TOKEN = os.getenv("HF_TOKEN")
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags")
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
 # Images
 torch.hub.download_url_to_file('https://i.pinimg.com/originals/7f/5e/96/7f5e9657c08aae4bcd8bc8b0dcff720e.jpg', 'ejemplo1.jpg')
@@ -50,6 +49,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)  # local model  o google colab
 #model = torch.hub.load('path/to/yolov5', 'custom', path='/content/yolov56.pt', source='local')  # local repo
 
+HF_TOKEN = os.getenv("HF_TOKEN")
+hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "demo-iazika-flags")
 
 def yolo(size, iou, conf, im):
     '''Wrapper fn for gradio'''