Update app.py

app.py

@@ -31,6 +31,8 @@ def generate(image_editor, prompt, neg_prompt, strength, guidance):
     image = image_editor['background'].convert('RGB')
     image.thumbnail((1024, 1024))
     image = divisible_by_8(image)
+
+    original_image_size = image.size
     
 
     layer = image_editor["layers"][0].resize(image.size)
@@ -56,6 +58,8 @@ def generate(image_editor, prompt, neg_prompt, strength, guidance):
                            #strength=strength, 
                            #guidance_scale=guidance).images[0]
 
+    
+    final_image = final_image.crop((0, 0, original_image_size.width, original_image_size.height))
     return image_editor, image, mask, final_image
 
 with gr.Blocks() as demo: