Update app.py

app.py

@@ -43,6 +43,7 @@ def greet(image):
 
     print(image.shape)
     print(mask_pred.shape)
+    print(mask_pred[0, 0, :, :].shape)
     return mask_pred[0, 0, :, :].numpy()