import gradio as gr
import matplotlib.pyplot as plt
from PIL import Image
from ultralytics import YOLO

model = YOLO('best (1).pt')

def response(image):
    results = model(image)
    # Plot results image
    im_bgr = results.plot()  # BGR-order numpy array
    im_rgb = im_bgr[..., ::-1]  # Convert BGR to RGB
    return im_rgb

iface = gr.Interface(fn=response, inputs="image", outputs="image")
iface.launch()