Update app.py

app.py

@@ -0,0 +1,25 @@
+import gradio as gr
+from ultralytics import YOLO
+import cv2
+import numpy as np
+# charger le modèle
+model = YOLO("yolov8n-seg.pt")
+
+# Fonction pour la détection sur image
+def detect_objects_image(img):
+    results = model(img)  # Détection
+    annotated_frame = results[0].plot()  # Annoter les résultats
+    return annotated_frame
+
+
+demo = gr.Blocks()
+
+#Interface Gradio
+image_input = gr.Image(type="numpy", label="Image à analyser")
+
+image_output = gr.Image(type="numpy", label="Image annotée")
+
+
+interface = gr.Interface(fn=detect_objects_image, inputs=image_input, outputs=image_output, title="Détection sur Image")
+
+interface.launch()