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, input="image", output="image")
iface.launch()