Fix: Add padding=True to image_processor to handle variable image sizes

app.py

@@ -94,7 +94,7 @@ def predict_pet():
         
         # Make prediction
         import torch
-        inputs = image_processor(pil_image, return_tensors="pt")
+        inputs = image_processor(pil_image, return_tensors="pt", padding=True)
         
         device = next(model.parameters()).device
         inputs = {k: v.to(device) for k, v in inputs.items()}