inflate mask before blur

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import numpy as np
 import spaces
 import torch
-from PIL import Image, ImageFilter
+from PIL import Image, ImageFilter, ImageOps
 from diffusers import FluxInpaintPipeline
 from gradio_client import Client, handle_file
 
@@ -138,6 +138,7 @@ def process(
             api_name="/process_image")
         mask = Image.open(mask)
 
+    mask = ImageOps.expand(mask, border=5, fill=255)
     mask = mask.filter(ImageFilter.GaussianBlur(radius=5))
     width, height = resize_image_dimensions(original_resolution_wh=image.size)
     image = image.resize((width, height), Image.LANCZOS)