import gradio as gr
import torch
from PIL import Image

model = torch.hub.load('ultralytics/yolov5', 'yolov5s') 

def detect_objects(image):
    results = model(image)
    return results.render()[0]

iface = gr.Interface(
    detect_objects,
    inputs=gr.Image(type="pil"),
    outputs=gr.Image(type="pil"),
    title="YOLO Object Detection",
    description="Upload an image to detect objects",
    examples=['./image_1.jpg','./image_2.jpg', '/image_3.jpg']
)

iface.launch()