properly select size

app.py

@@ -35,8 +35,6 @@ import torch
 from PIL import Image
 import requests
 
-url = "http://images.cocodataset.org/val2017/000000039769.jpg"
-image = Image.open(requests.get(url, stream=True).raw)
 
 processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
 model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
@@ -84,7 +82,7 @@ def streaming_object_detection(recording_id: str, img):
     # convert outputs (bounding boxes and class logits) to COCO API
     # let's only keep detections with score > 0.9
 
-    target_sizes = torch.tensor([img.height, img.width])
+    target_sizes = torch.tensor([img.size[::-1]])
     results = processor.post_process_object_detection(
         outputs, target_sizes=target_sizes, threshold=0.9
     )[0]