Fix mask

app.py

@@ -7,7 +7,7 @@ import spaces
 import torch
 from diffusers import FluxFillPipeline
 from gradio.components.image_editor import EditorValue
-from PIL import Image, ImageOps
+from PIL import Image, ImageFilter, ImageOps
 
 DEVICE = "cuda"
 
@@ -122,12 +122,16 @@ def infer(
         (255, 255, 255),
     )
     mask.paste(_furniture_mask, (0, 0))
-    mask.paste(_room_mask, (FIXED_DIMENSION, 0))
+    mask.paste(_room_mask, (FIXED_DIMENSION, 0), _room_mask)
 
     width, height = calculate_optimal_dimensions(image)
     # Resize the image and mask to the optimal dimensions for the VAe
     image = image.resize((width, height))
     mask = mask.resize((width, height))
+    # Invert the mask
+    mask = ImageOps.invert(mask)
+    # Blur the mask
+    mask = mask.filter(ImageFilter.GaussianBlur(radius=33))
 
     if randomize_seed:
         seed = secrets.randbelow(MAX_SEED)
@@ -183,7 +187,7 @@ with gr.Blocks(css=css) as demo:
                         label="Room Image - Draw mask for inpainting",
                         type="pil",
                         sources=["upload"],
-                        image_mode="RGB",
+                        image_mode="RGBA",
                         layers=False,
                         brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"),
                         height=300,