sda

app.py

@@ -3,9 +3,11 @@ import gradio as gr
 from transformers import pipeline
 from PIL import Image
 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", device=-1)
+detector = pipeline("object-detection", model="facebook/detr-resnet-50").to(device)
 
 def process_video(video_path):
     """