Files

app.py

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