Upload 3 files

app.py

@@ -0,0 +1,21 @@
+import cv2
+import gradio as gr
+from ultralytics import YOLO
+
+model = YOLO("yolov5n.pt")
+
+def predict(img):
+
+    pred = model(img)
+    frame_with_bbox = pred[0].plot()
+    return frame_with_bbox
+
+
+
+gradio_app = gr.Interface(
+             fn=predict,
+             inputs=gr.Image(type="numpy"),
+             outputs=gr.Image(type="numpy"))
+
+if __name__ == "__main__":
+    gradio_app.launch()

requirements.txt

@@ -0,0 +1,2 @@
+opencv-python
+ultralytics

yolov5n.pt

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