Update app.py

app.py

@@ -5,11 +5,8 @@ import gradio as gr
 from PIL import Image
 from transformers import Owlv2Processor, Owlv2ForObjectDetection
 
-model_id = "IDEA-Research/grounding-dino-base"
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
-processor = AutoProcessor.from_pretrained(model_id)
-model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)
+processor = Owlv2Processor.from_pretrained("google/owlv2-large-patch14-finetuned")
+model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-large-patch14-finetuned")
 
 def input_image_setup(uploaded_file):
     """