Update app.py

app.py

@@ -35,13 +35,16 @@ def generate(image_editor, prompt, neg_prompt, strength, guidance):
 
     layer = image_editor["layers"][0].resize(image.size)
 
-    image = squarify_image(image)
-    layer = squarify_image(layer).convert('RGBA')
+    
     
     
     mask = Image.new("RGBA", image.size, "WHITE") 
     mask.paste(layer, (0, 0), layer)
     mask = ImageOps.invert(mask.convert('L'))
+
+
+    image = squarify_image(image)
+    layer = squarify_image(layer)
     
 
     final_image = pipeline(prompt=prompt,