Update app.py

app.py

@@ -6,14 +6,31 @@ from PIL import Image, ImageOps
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipeline = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to("cuda")
 
+
+def divisible_by_8(image):
+    width, height = image.size
+    
+    # Calculate the new width and height that are divisible by 8
+    new_width = (width // 8) * 8
+    new_height = (height // 8) * 8
+    
+    # Resize the image
+    resized_image = image.resize((new_width, new_height))
+    
+    return resized_image
+
+
 def generate(image_editor, prompt, neg_prompt, strength, guidance):
     image = image_editor['background'].convert('RGB')
-    mask = Image.new("RGBA", image_editor["layers"][0].size, "WHITE") 
-    mask.paste(image_editor["layers"][0], (0, 0), image_editor["layers"][0])
+    image.thumbnail((1024, 1024))
+    image = divisible_by_8(image)
+
+    layer = image_editor["layers"][0].resize(image.size)
+    
+    mask = Image.new("RGBA", image.size, "WHITE") 
+    mask.paste(layer, (0, 0), layer)
     mask = ImageOps.invert(mask.convert('L'))
     
-    image.thumbnail((1024, 1024))
-    mask.thumbnail((1024, 1024))
 
     final_image = pipeline(prompt=prompt, 
                            image=image,