Create app.py

app.py

@@ -0,0 +1,14 @@
+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()