Update app.py

app.py

@@ -66,6 +66,9 @@ def inference_mask1(
     query_img = query_img.unsqueeze(0)
     support_img = torch.stack(support_imgs, dim=0).unsqueeze(0)
     support_masks = torch.stack(support_masks, dim=0).unsqueeze(0)
+    print("query_img:", query_img.size())
+    print("support_img:", support_img.size())
+    print("support_masks:", support_masks.size())
     batch = {
         "support_masks": support_masks,
         "support_imgs": support_img,