Update app.py

app.py

@@ -11,7 +11,7 @@ model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
 def predict_image(image: Image.Image):
     try:
         # Step 1: Use DETR for object detection
-        inputs = processor(images=image, return_tensors="pt")
+        inputs = processor(images=image, return_tensors="pt", padding=True)
         
         with torch.no_grad():
             outputs = model(**inputs)