Update app.py

app.py

@@ -1,8 +1,26 @@
+import cv2
 import gradio as gr
+from ultralytics import YOLO
 
-def greet(img):
-    print(img)
-    return "Hello World"
+def process_frame(model, frame):
+    results = model(frame)
+    for result in results:
+        for box in result.boxes.data.tolist():
+            x1, y1, x2, y2, confidence, class_id = box
+            label = f"{model.names[int(class_id)]} {confidence:.2f}"
+            cv2.rectangle(frame, (int(x1), int(y1)), (int(x2), int(y2)), (0, 255, 0), 3)
+            cv2.putText(frame, label, (int(x1), int(y1) - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 255, 0), 2)
+    return frame
+
+def detect_objects(image):
+    model = YOLO("s.pt")  # Путь к вашей модели
+    processed_image = process_frame(model, image)
+    return processed_image
+
+# Интерфейс Gradio
+iface = gr.Interface(fn=detect_objects, 
+                     inputs=gr.Image(type="numpy"), 
+                     outputs=gr.Image(type="numpy"),
+                     live=True)
 
-iface = gr.Interface(fn=greet, inputs="image", outputs="text")
 iface.launch()