Add shape attribute to the array object

[jivaniyash]

watermark-removal/preprocess_image.py

Line 42 in 725b762

assert image.shape == preprocessed_mask_image

Please change the line to

assert image.shape == preprocessed_mask_image.shape`