Update app.py

app.py

@@ -81,7 +81,7 @@ def query(image, confidence_threshold):
   inputs = processor(images=image, return_tensors="pt").to(device)
   with torch.no_grad():
     outputs = model(**inputs)
-  target_sizes = torch.Tensor([image.size])
+  target_sizes = torch.Tensor([np.array(image).shape[:-1]])
   
   results = processor.post_process_object_detection(outputs=outputs, threshold=confidence_threshold, target_sizes=target_sizes)
   return results