Update app.py

app.py

@@ -20,7 +20,8 @@ def query_image(img, text_queries, score_threshold):
     text_queries = text_queries
     text_queries = text_queries.split(",")
 
-    target_sizes = torch.Tensor([img.shape[:2]])
+    size = max(img.shape[:2])
+    target_sizes = torch.Tensor([[size, size]])
     inputs = processor(text=text_queries, images=img, return_tensors="pt").to(device)
 
     with torch.no_grad():