Update app.py

app.py

@@ -14,8 +14,8 @@ def predict(image):
         if isinstance(image, str):
             image = Image.open(image)
         
-        # Preprocess the input image using the processor
-        inputs = processor(images=image, return_tensors="pt")
+        # Preprocess the input image using the processor, with padding enabled
+        inputs = processor(images=image, return_tensors="pt", padding=True)
         
         # Get the model's predictions
         with torch.no_grad():