asd

app.py

@@ -6,8 +6,26 @@ import tempfile
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-# Cargar el modelo de detección de objetos usando CPU
-detector = pipeline("object-detection", model="facebook/detr-resnet-50").to(device)
+print(f"Using device: {device}")
+
+# Cargar el modelo de detección de objetos
+try:
+    detector = pipeline(
+        "object-detection",
+        model="facebook/detr-resnet-50",
+        device=0 if device == "cuda" else -1,  # 0 para GPU, -1 para CPU
+        framework="pt"  # Especificar PyTorch como framework
+    )
+    print("Model loaded successfully on", device)
+except Exception as e:
+    print(f"Error loading model: {e}")
+    print("Falling back to CPU")
+    detector = pipeline(
+        "object-detection",
+        model="facebook/detr-resnet-50",
+        device=-1,
+        framework="pt"
+    )
 
 def process_video(video_path):
     """