Update app.py

app.py

@@ -32,15 +32,17 @@ def greet(image):
         A.Resize(height=512, width=512),
         ToTensorV2(),
     ])
-
+    print(image.shape)
     image = inference_transforms(image=image)["image"]
-
+    print(image.shape)
     image = image.unsqueeze(0)
 
 
     with torch.no_grad():
         mask_pred = torch.sigmoid(model(image))
-        
+
+    print(image.shape)
+    print(mask_pred.shape)
     return mask_pred[0].numpy()