Update app.py

app.py

@@ -104,8 +104,8 @@ def process_image_detection(image, target_label, surprise_rating):
     # Calculate relative font size based on image dimensions
     base_fontsize = min(original_size) / 40  # Adjust this divisor to change overall font size
 
-    owlv2_processor = Owlv2Processor.from_pretrained("google/owlv2-base-patch14")
-    owlv2_model = Owlv2ForObjectDetection.from_pretrained("google/owlv2-base-patch14").to(device)
+    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)