Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import AutoImageProcessor, AutoModelForObjectDetection
+from PIL import Image
 import torch
 
 image_processor = AutoImageProcessor.from_pretrained('hustvl/yolos-small')
@@ -8,6 +9,7 @@ model = AutoModelForObjectDetection.from_pretrained('hustvl/yolos-small')
 def detect(image): 
     inputs = image_processor(images=image, return_tensors="pt")
     outputs = model(**inputs)
+    image = Image.open(image)
 
     # convert outputs to COCO API
     target_sizes = torch.tensor([image.size[::-1]])