Update app.py

app.py

@@ -181,10 +181,10 @@ def process_image_detection(image, target_label, surprise_rating):
        base_fontsize = min(original_size) / 80
 
        print("Loading models...")
-       owlv2_processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch16")
-       owlv2_model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch16").to(device)
-       sam_processor = AutoProcessor.from_pretrained("facebook/sam-vit-base")
-       sam_model = AutoModelForMaskGeneration.from_pretrained("facebook/sam-vit-base").to(device)
+       owlv2_processor = Owlv2Processor.from_pretrained("google/owlv2-large-patch14")
+       owlv2_model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-large-patch14").to(device)
+       sam_processor = AutoProcessor.from_pretrained("facebook/sam-vit-large")
+       sam_model = AutoModelForMaskGeneration.from_pretrained("facebook/sam-vit-large").to(device)
 
        print("Running object detection...")
        inputs = owlv2_processor(text=[target_label], images=image, return_tensors="pt").to(device)