Update app.py

app.py

@@ -1,6 +1,6 @@
 import torch
 import gradio as gr
-from transformers import OwlViTProcessor, OwlViTForObjectDetection
+from transformers import Owlv2Processor, Owlv2ForObjectDetection
 
 
 # Use GPU if available
@@ -9,9 +9,8 @@ if torch.cuda.is_available():
 else:
     device = torch.device("cpu")
 
-model = OwlViTForObjectDetection.from_pretrained("nielsr/owlv2-base-patch16-ensemble").to(device)
-model.eval()
-processor = OwlViTProcessor.from_pretrained("nielsr/owlv2-base-patch16-ensemble")
+model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16-ensemble").to(device)
+processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16-ensemble")
 
 
 def query_image(img, text_queries, score_threshold):