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):
    for i, r in enumerate(results):
    # Plot results image
    im_bgr = r.plot()  # BGR-order numpy array
    im_rgb = im_bgr[..., ::-1]  # Convert BGR to RGB
    plt.figure(figsize=(10, 10))
    plt.imshow(im_rgb)
    plt.axis('off')
    plt.show()
    return results

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