app.py

import gradio as gr

import torch

model = torch.hub.load("ultralytics/yolov5", "best")

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()