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):
    img = Image.fromarray(image)
    results = model(img)
    # Plot results image
    
    return results

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