Files

app.py

@@ -7,8 +7,8 @@ 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("google/owlv2-base-patch16-ensemble").to("cuda")
+owl_processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
 
 # owl_processor = AutoProcessor.from_pretrained("IDEA-Research/grounding-dino-base")
 # owl_model = AutoModelForZeroShotObjectDetection.from_pretrained("IDEA-Research/grounding-dino-base").to("cuda")