resize

app.py

@@ -16,11 +16,17 @@ def create_mask(input):
     
     input.save(input_path)
     bg_removed = remove(input)
+
+    width, height = bg_removed.size
+    max_dim = max(width, height)
+    square_img = Image.new('RGB', (max_dim, max_dim), (255, 255, 255))
+    paste_pos = ((max_dim - width) // 2, (max_dim - height) // 2)
+    square_img.paste(bg_removed, paste_pos)
     
-    bg_removed = bg_removed.resize((512, 512))
-    bg_removed.save(bg_removed_path)
+    square_img = square_img.resize((512, 512))
+    square_img.save(bg_removed_path)
 
-    img2_grayscale = bg_removed.convert('L')
+    img2_grayscale = square_img.convert('L')
     img2_a = np.array(img2_grayscale)
 
     mask = np.array(img2_grayscale)