move to gpu

owl_core.py

@@ -46,8 +46,8 @@ def owl_full_video(
         threshold: float,
         fps_processed: int = 1,
         scaling_factor: float = 0.5,
-        processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble"),
-        model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble"),
+        processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble").to('cuda'),
+        model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to('cuda'),
         device: str = 'cuda',
         batch_size: int = 6,
         ):