Update app.py

app.py

@@ -5,8 +5,8 @@ import gradio as gr
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-owl_model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16").to("cuda")
-owl_processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16")
+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("cuda")