Update app.py

app.py

@@ -5,11 +5,32 @@ from ultralyticsplus import YOLO, render_result
 import cv2
 import numpy as np
 from transformers import pipeline
+import requests
+from io import BytesIO
 
 model = YOLO('best (1).pt')
 model2 = pipeline('image-classification','Kaludi/csgo-weapon-classification')
 name = ['grenade','knife','pistol','rifle']
 
+url_example="https://drive.google.com/file/d/1bBq0bNmJ5X83tDWCzdzHSYCdg-aUL4xO/view?usp=drive_link"
+url_example='https://drive.google.com/uc?id=' + url_example.split('/')[-2]
+r = requests.get(url_example)
+im1 = Image.open(BytesIO(r.content))
+
+url_example="https://drive.google.com/file/d/16Z7QzvZ99fbEPj1sls_jOCJBsC0h_dYZ/view?usp=drive_link"
+url_example='https://drive.google.com/uc?id=' + url_example.split('/')[-2]
+r = requests.get(url_example)
+im2 = Image.open(BytesIO(r.content))
+
+url_example="https://drive.google.com/file/d/13mjTMS3eR0AKYSbV-Fpb3fTBno_T42JN/view?usp=drive_link"
+url_example='https://drive.google.com/uc?id=' + url_example.split('/')[-2]
+r = requests.get(url_example)
+im3 = Image.open(BytesIO(r.content))
+
+url_example="https://drive.google.com/file/d/13mjTMS3eR0AKYSbV-Fpb3fTBno_T42JN/view?usp=drive_link"
+url_example='https://drive.google.com/uc?id=' + url_example.split('/')[-2]
+r = requests.get(url_example)
+im4 = Image.open(BytesIO(r.content))
  # for i, r in enumerate(results):
       
  #    # Plot results image
@@ -92,11 +113,11 @@ outputs = [gr.Image( type="filepath", label="Output Image"),
           ]
 
 
-# examples = [['th (11).jpg', 640, 0.3, 0.6],
-#             ['th (8).jpg', 640, 0.3, 0.6],
-#             ['th (3).jpg', 640, 0.3, 0.6],
-#             ['th.jpg', 640, 0.15, 0.6]
-#            ]
+examples = [[im1, 640, 0.3, 0.6],
+            [im2, 640, 0.3, 0.6],
+            [im3, 640, 0.3, 0.6],
+            [im4, 640, 0.15, 0.6]
+           ]