Update app.py

app.py

@@ -31,10 +31,13 @@ def generate(image_editor, prompt, neg_prompt, strength, guidance):
     image = image_editor['background'].convert('RGB')
     image.thumbnail((1024, 1024))
     image = divisible_by_8(image)
-    image = squarify_image(image)
+    
 
     layer = image_editor["layers"][0].resize(image.size)
-    layer = squarify_image(layer)
+
+    image = squarify_image(image)
+    layer = squarify_image(layer).convert('RGBA')
+    
     
     mask = Image.new("RGBA", image.size, "WHITE") 
     mask.paste(layer, (0, 0), layer)