import gradio as gr

from ultralytics import YOLO

model = YOLO("best.pt")

def greet(source):
  model = YOLO("best.pt")
  results = model.predict(source, imgsz=1080, conf=0.25)
  result = results[0]
  return result.plot()

demo = gr.Interface(fn=greet, inputs="image", outputs="image")
demo.launch()