Upload app.py

app.py

@@ -8,8 +8,9 @@ import numpy as np
 # Modèles optimisés pour le temps réel
 REALTIME_MODELS = {
     "YOLOS Tiny (ultra-rapide)": "hustvl/yolos-tiny",
+    "YOLOS Small": "hustvl/yolos-small",
     "DETR ResNet-50": "facebook/detr-resnet-50",
-    "YOLOS Small": "hustvl/yolos-small"
+    "Conditional DETR (optimisé)": "microsoft/conditional-detr-resnet-50"
 }
 
 # Variables globales
@@ -86,10 +87,15 @@ def process_webcam(image, model_choice, confidence_threshold):
         detections = detector(resized_image)
         print(f"🎯 Détections brutes: {len(detections)}")
         
+        # Debug: afficher le format des détections pour DETR
+        if detections and model_choice == "DETR ResNet-50":
+            print(f"🔧 Debug DETR - Première détection: {detections[0]}")
+            print(f"🔧 Keys disponibles: {list(detections[0].keys())}")
+        
         # Filtrer par confiance
         filtered_detections = [
             det for det in detections 
-            if det['score'] >= confidence_threshold
+            if det.get('score', 0) >= confidence_threshold
         ]
         
         print(f"✅ Détections filtrées: {len(filtered_detections)}")
@@ -151,7 +157,7 @@ def draw_detections(image, detections):
     
     return img_copy
 
-# Interface avec configuration webcam améliorée
+# Interface avec activation webcam forcée
 demo = gr.Interface(
     fn=process_webcam,
     inputs=[
@@ -159,8 +165,8 @@ demo = gr.Interface(
             sources=["webcam"], 
             streaming=True, 
             type="pil",
-            mirror_webcam=False,
-            show_download_button=False
+            show_download_button=False,
+            interactive=True
         ),
         gr.Dropdown(
             choices=list(REALTIME_MODELS.keys()),
@@ -172,8 +178,7 @@ demo = gr.Interface(
     outputs=gr.Image(streaming=True, type="pil", show_download_button=False),
     live=True,
     title="🎥 Détection Live",
-    description="⚠️ Cliquez sur l'icône webcam et autorisez l'accès pour commencer",
-    allow_flagging="never",
+    description="🚨 IMPORTANT: Activez la webcam en cliquant sur l'icône caméra dans la zone Image",
     flagging_mode="never"
 )