import gradio as gr
from ultralytics import YOLO
from PIL import Image

det_model = YOLO("yolov8n.pt")

def detect_objects(image):
    results = det_model(image)
    det_img = results[0].plot()  
    return det_img


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

iface.launch()