Sample video path added

app.py

@@ -8,7 +8,8 @@ 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'
+    'https://www.dropbox.com/s/86uxlxxlm1iaexa/pothole_screenshot.png?dl=1',
+    'https://www.dropbox.com/s/7sjfwncffg8xej2/video_7.mp4?dl=1'
 ]
 
 def download_file(url, save_name):
@@ -17,14 +18,21 @@ def download_file(url, 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"
-    )
+for i, url in enumerate(file_urls):
+    if 'mp4' in file_urls[i]:
+        download_file(
+            file_urls[i],
+            f"video.mp4"
+        )
+    else:
+        download_file(
+            file_urls[i],
+            f"image_{i}.jpg"
+        )
 
 model = YOLO('best.pt')
 path  = [['image_0.jpg'], ['image_1.jpg']]
+video_path = [['video.mp4']]
 
 def show_preds_image(image_path):
     image = cv2.imread(image_path)
@@ -93,7 +101,7 @@ interface_video = gr.Interface(
     inputs=inputs_video,
     outputs=outputs_video,
     title="Pothole detector",
-    # examples=path,
+    examples=video_path,
     cache_examples=False,
     # live=True,
 )