Update app.py

app.py

@@ -13,8 +13,24 @@ pipe_prior.to("cuda")
 pipe = KandinskyPipeline.from_pretrained("kandinsky-community/kandinsky-2-1", torch_dtype=torch.float16)
 pipe.to("cuda")
 
+
+@spaces.GPU()
+def squarify_image(img):
+    if(img.height > img.width): bg_size = img.height
+    else:  bg_size = img.width
+    bg = Image.new(mode="RGB", size=(bg_size,bg_size), color="white")
+    bg.paste(img, ( int((bg.width - bg.width)/2), 0) )
+
+    return bg
+
 @spaces.GPU()
 def blend(img1, img2, slider):
+    img1.thumbnail((1024, 1024))
+    img2.thumbnail((1024, 1024))
+    
+    img1 = squarify_image(img1)
+    img2 = squarify_image(img2)
+    
     # add all the conditions we want to interpolate, can be either text or image
     images_texts = [img1, img2]