Upload 3 files

app.py

@@ -0,0 +1,23 @@
+import torch
+import ultralytics
+from ultralytics import YOLO
+import gradio as gr
+import cv2
+
+# Загружаем веса, которые лежат в репозитории
+model = YOLO("best.pt")
+
+def detect(image):
+    results = model.predict(image, imgsz=640, conf=0.1, verbose=False)
+    bgr = results[0].plot()
+    rgb = cv2.cvtColor(bgr, cv2.COLOR_BGR2RGB)
+    return rgb
+
+demo = gr.Interface(
+    fn=detect,
+    inputs=gr.Image(type="numpy", label="Upload image"),
+    outputs=gr.Image(type="numpy", label="Detection result"),
+)
+
+if __name__ == "__main__":
+    demo.launch()

best.pt

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