Update app.py

app.py

@@ -7,7 +7,8 @@ from ultralytics import YOLO
 
 file_urls = ["https://www.kp-glas.de/wp-content/uploads/2023/02/Bierflaschen.jpeg", 
              "https://www.bevindustry.com/ext/resources/issues/2020/February/Heineken-0-0-Alcohol-Free-Beer.jpg?1581450636",
-            "https://spice-world.co.za/cdn/shop/products/3520_648ad722263c64.28810246_drinks_20copy_1db265d5-e818-45d3-973b-3e046a6b0931_800x.jpg?v=1697124785"]
+            "https://spice-world.co.za/cdn/shop/products/3520_648ad722263c64.28810246_drinks_20copy_1db265d5-e818-45d3-973b-3e046a6b0931_800x.jpg?v=1697124785",
+            "https://i.redd.it/0e2zml4mwz021.jpg"]
 
 def download_file(url, save_name):
     if not os.path.exists(save_name):
@@ -15,10 +16,7 @@ def download_file(url, save_name):
         open(save_name, 'wb').write(file.content)
 
 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")
+    download_file(file_urls[i], f"image_{i}.jpg")
 
 # Available model options
 model_paths = [
@@ -27,7 +25,7 @@ model_paths = [
     ]
 
 # Paths to example images
-path = [['image_0.jpg'], ['image_1.jpg'], ['image_2.jpg']]
+path = [['image_0.jpg'], ['image_1.jpg'], ['image_2.jpg'],  ['image_3.jpg']]
 
 # Function to run inference and show predictions
 def show_preds_image(image_path, selected_model):