center crop

app.py

@@ -16,6 +16,15 @@ def create_mask(input):
     
     input.save(input_path)
     bg_removed = remove(input)
+
+    width, height = bg_removed.size
+    new_size = min(width, height)
+    left = (width - new_size) // 2
+    top = (height - new_size) // 2
+    right = (width + new_size) // 2
+    bottom = (height + new_size) // 2
+    bg_removed = bg_removed.crop((left, top, right, bottom))
+    
     bg_removed = bg_removed.resize((512, 512))
     bg_removed.save(bg_removed_path)