added gradio demo

app.py

@@ -0,0 +1,27 @@
+from ultralytics import YOLO
+from ultralytics.utils.plotting import Annotator
+import torch
+import gradio as gr
+import cv2
+model = YOLO('best.pt')
+def yolo_pred(image):
+    result = model.predict(image)[0]
+    annotator = Annotator(result.orig_img)
+    color_list = [(107, 31, 45), (32, 102, 50), (32, 45, 102)]
+    for label in result.boxes.data.detach().numpy():
+        annotator.box_label(
+            label[0:4],
+            str(result.names[label[-1].item()]) + " " + str(round(label[-2], 2)),
+            color_list[int(label[-1].item())]
+        )
+        print(round(label[-2], 2))
+    return annotator.im
+gr.Interface(fn=yolo_pred, 
+             inputs="image", 
+             outputs="image",
+             examples=[
+                 [cv2.imread("example1.jpg")],
+                 [cv2.imread("example2.jpg")],
+                 [cv2.imread("example3.jpg")],
+             ]
+            ).launch()

best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf47a92c317e706a8bfbea42a92455dbae93c500c373471f044487ba685faedc
+size 6262115