resize to 512

app.py

@@ -15,8 +15,12 @@ def image_to_byte_array(image: Image) -> bytes:
     return imgByteArr
 
 def predict(input_img1,input_img2):
+
     print("predict")
 
+    input_img1 = input_img1.resize((512, 512))
+    input_img2 = input_img2.resize((512, 512))
+
     canvas_mask = np.array(input_img2)
     mask = get_mask(canvas_mask)