Fix bug and new samples

app.py

@@ -1,14 +1,8 @@
 import gradio as gr
-import torch
+from PIL import Image
+import numpy as np
 from ultralytics import YOLO
-
-torch.hub.download_url_to_file(
-    'http://www.imesclub.org/images/stories/arabidentity.jpeg', 'one.jpg')
-torch.hub.download_url_to_file(
-    'https://lp-cms-production.imgix.net/2023-08/GettyImages-1224548888-16.9.jpg', 'two.jpg')
-torch.hub.download_url_to_file(
-    'https://s.wsj.net/public/resources/images/OB-EQ974_diwali_G_20091014112400.jpg', 'three.jpg')
-
+import os
 
 def handle_classify(image=None):
     """This function performs YOLOv8 object detection on the given image.
@@ -36,7 +30,7 @@ def handle_classify(image=None):
 
 
 inputs = [
-    gr.Image(type="filepath", label="Input Image"),
+    gr.Image(type='numpy', label="Input Image"),
 ]
 
 
@@ -44,10 +38,8 @@ outputs = gr.Textbox()
 
 title = "Racist model v2"
 
-
-examples = [['one.jpg'],
-            ['two.jpg'],
-            ['three.jpg']]
+SAMPLE_DIR = 'samples'
+examples = [np.array(Image.open(os.path.join(SAMPLE_DIR, path))) for path in os.listdir(SAMPLE_DIR)]
 
 yolo_app = gr.Interface(
     fn=handle_classify,

requirements.txt

@@ -1,3 +1,4 @@
 torch
 gradio==4.14.0
-ultralytics
+ultralytics
+PIL