Update app.py

app.py

@@ -5,11 +5,11 @@ import gradio as gr
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-owl_model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to(device)
+owl_model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to("cuda")
 owl_processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
 
 dino_processor = AutoProcessor.from_pretrained("IDEA-Research/grounding-dino-base")
-dino_model = AutoModelForZeroShotObjectDetection.from_pretrained("IDEA-Research/grounding-dino-base").to(device)
+dino_model = AutoModelForZeroShotObjectDetection.from_pretrained("IDEA-Research/grounding-dino-base").to("cuda")
 
 @spaces.GPU
 def infer(img, text_queries, score_threshold, model):