Images download from URL

app.py

@@ -1,11 +1,30 @@
 import gradio as gr
 from gradio.outputs import Label
 import cv2
+import requests
+import os
 
 from ultralytics import YOLO
 
+file_urls = [
+    'https://www.dropbox.com/s/b5g97xo901zb3ds/pothole_example.jpg?dl=1',
+    'https://www.dropbox.com/s/86uxlxxlm1iaexa/pothole_screenshot.png?dl=1'
+]
+
+def download_file(url, save_name):
+    url = url
+    if not os.path.exists(save_name):
+        file = requests.get(url)
+        open(save_name, 'wb').write(file.content)
+
+for i, url in enumerate(file_urls):    
+    download_file(
+        file_urls[i],
+        f"image_{i}.jpg"
+    )
+
 model = YOLO('best.pt')
-path  = [['pothole_example.jpg'], ['pothole_screenshot.png']]
+path  = [['image_0.jpg'], ['image_1.jpg']]
 
 def show_preds(image_path):
     image = cv2.imread(image_path)
@@ -30,8 +49,8 @@ gr_interface = gr.Interface(
     outputs=gr.outputs.Image(type="numpy", label="Output Image"),
     title="Pothole detector",
     examples=path,
-    cache_examples=True,
-    live=True,
+    # cache_examples=True,
+    # live=True,
 )
 
 gr_interface.launch(debug=True, enable_queue=True)

gradio_cached_examples/12/log.csv

@@ -0,0 +1,3 @@
+Output Image,flag,username,timestamp
+/media/sovit/hdd1tbinternal/my_data/Data_Science/Big_Vision/pothole_project/huggingface_spaces/yolov8n/pothole_yolov8_nano/gradio_cached_examples/12/Output Image/tmp2gxdj84j.png,,,2023-01-13 09:31:17.531032
+/media/sovit/hdd1tbinternal/my_data/Data_Science/Big_Vision/pothole_project/huggingface_spaces/yolov8n/pothole_yolov8_nano/gradio_cached_examples/12/Output Image/tmpz66236qf.png,,,2023-01-13 09:31:18.272697