diff --git a/visual_workflow_builder/backend/services/ui_detection_service.py b/visual_workflow_builder/backend/services/ui_detection_service.py
index 1c296df17..097978ef0 100644
--- a/visual_workflow_builder/backend/services/ui_detection_service.py
+++ b/visual_workflow_builder/backend/services/ui_detection_service.py
@@ -108,10 +108,12 @@ def _load_rfdetr():
         return _rfdetr_model
 
     from rfdetr.detr import RFDETRMedium
-    print(f"[UI-DETR-1] Chargement du modèle...")
+    import torch
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    print(f"[UI-DETR-1] Chargement du modèle sur {device}...")
     start = time.time()
-    _rfdetr_model = RFDETRMedium(pretrain_weights=MODEL_PATH, resolution=RESOLUTION)
-    print(f"[UI-DETR-1] Modèle chargé en {time.time() - start:.1f}s")
+    _rfdetr_model = RFDETRMedium(pretrain_weights=MODEL_PATH, resolution=RESOLUTION, device=device)
+    print(f"[UI-DETR-1] Modèle chargé en {time.time() - start:.1f}s ({device})")
     return _rfdetr_model