Update app.py

app.py

@@ -111,6 +111,12 @@ def outpaintGen(
     # Paste white rectangle at the appropriate position
     mask_image.paste(white_area, (overlap_left, overlap_top))
 
+    # Create a new image with the same size as the original image
+    new_image = Image.new('RGB', (new_width, new_height), color='black')
+
+    # Paste the original image onto the new image
+    new_image.paste(image, (overlap_left, overlap_top))
+
     # Convert to grayscale
     mask_image = mask_image.convert('L')
     mask_image = Image.eval(mask_image, lambda x: 255 - x)
@@ -128,7 +134,7 @@ def outpaintGen(
 
     result = pipe(
         prompt=outpaint_prompt,
-        image=image,
+        image=new_image,
         mask_image=mask_image,
         width=fix_width,
         height=fix_height,