crop image to square

app.py

@@ -66,23 +66,16 @@ i2i_pipe.to(device=torch_device, dtype=torch_dtype).to(device)
 i2i_pipe.set_progress_bar_config(disable=True)
 
 
-def pad_image(image):
+def resize_crop(image, size=512):
+    image = image.convert("RGB")
     w, h = image.size
-    if w == h:
-        return image
-    elif w > h:
-        new_image = Image.new(image.mode, (w, w), (0, 0, 0))
-        new_image.paste(image, (0, (w - h) // 2))
-        return new_image
-    else:
-        new_image = Image.new(image.mode, (h, h), (0, 0, 0))
-        new_image.paste(image, ((h - w) // 2, 0))
-        return new_image
+    image = image.resize((size, int(size * (h / w))), Image.BICUBIC)
+    return image
 
 
 async def predict(init_image, prompt, strength, steps, seed=1231231):
     if init_image is not None:
-        init_image = pad_image(init_image).convert("RGB").resize((512, 512))
+        init_image = resize_crop(init_image)
         generator = torch.manual_seed(seed)
         last_time = time.time()
         results = i2i_pipe(