Spaces:

caltech-animal-tracking
/

Primate_Detection_V2

Sleeping

annayding commited on Dec 10, 2024

Commit

f024a1d

1 Parent(s): 47342ef

processor device

Files changed (1) hide show

owl_core.py CHANGED Viewed

@@ -46,7 +46,7 @@ def owl_full_video(
         threshold: float,
         fps_processed: int = 1,
         scaling_factor: float = 0.5,
-        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,

         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").to('cuda'),
         device: str = 'cuda',
         batch_size: int = 6,