resize image to 512x512

app.py

@@ -11,6 +11,18 @@ import cv2
 def create_key(seed=0):
     return jax.random.PRNGKey(seed)
 
+def crop_and_resize(pilimg, size=512):
+    """
+    Will downsample or upsample as necessary.
+    """
+    width, height = pilimg.size
+    minsize = min(width, height)
+    x0 = (width - height) // 2 if width > height else 0
+    y0 = (height - width) // 2 if height > width else 0
+    pilimg = pilimg.crop((x0, y0, x0 + minsize, y0 + minsize))
+    pilimg = pilimg.resize((size, size), resample=Image.LANCZOS)
+    return pilimg
+
 def canny_filter(image):
     gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     blurred_image = cv2.GaussianBlur(gray_image, (5, 5), 0)
@@ -27,6 +39,11 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
 
 def infer(prompts, negative_prompts, image):
     params["controlnet"] = controlnet_params
+
+    # image is a numpy array, we'll convert to PIL to resize and back to numpy
+    image = Image.fromarray(image)
+    image = crop_and_resize(image)
+    image = np.array(image)
     
     num_samples = 1 #jax.device_count()
     rng = create_key(0)