Add mask blur

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import torch
 from diffusers import FluxFillPipeline
 from diffusers.utils import load_image
-from PIL import Image, ImageDraw
+from PIL import Image, ImageDraw, ImageFilter
 import numpy as np
 import spaces
 from huggingface_hub import hf_hub_download
@@ -141,7 +141,8 @@ def inpaint(image, width, height, overlap_percentage, num_inference_steps, resiz
     ).images[0]
    
     result = result.convert("RGBA")
-    cnet_image.paste(result, (0, 0), mask)
+    mask_blur = mask.filter(ImageFilter.GaussianBlur(10))
+    cnet_image.paste(result, (0, 0), mask_blur)
 
     return cnet_image, background