Update app.py

app.py

@@ -1,11 +1,11 @@
+# app.py
 import gradio as gr
 import cv2
 from ultralytics import YOLO
 import tempfile
-import os
 
-# Charger le modèle
-model = YOLO("./yolo12m.pt")  # ou yolov8n.pt
+# Charger le modèle YOLO (chemin relatif)
+model = YOLO("./yolo12m.pt")
 
 # Classes à détecter
 target_classes = ["backpack", "suitcase", "handbag"]
@@ -26,7 +26,7 @@ def process_video(video_path):
     class_counts = {c: 0 for c in target_classes}
     total_count = 0
 
-    # Output temporaire
+    # Vidéo temporaire de sortie
     temp_out = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(temp_out.name, fourcc, 20, (W, H))
@@ -66,7 +66,7 @@ def process_video(video_path):
             cv2.putText(frame, f"{name}: {class_counts[name]}", (20, y_offset),
                         cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 255, 255), 2)
             y_offset += 30
-        cv2.putText(frame, f"Total: {total_count}", (20, y_offset + 10),
+        cv2.putText(frame, f"Total bagages : {total_count}", (20, y_offset + 10),
                     cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 0, 0), 2)
         cv2.line(frame, (0, line_y), (W, line_y), (0, 0, 255), 2)