Set mask to 1 channel

app.py

@@ -118,13 +118,15 @@ def infer(
     # Invert the mask
     mask = ImageOps.invert(mask)
     # Blur the mask
-    mask = mask.filter(ImageFilter.GaussianBlur(radius=33))
+    mask = mask.filter(ImageFilter.GaussianBlur(radius=10))
+    # Convert to 1 channel
+    mask = mask.convert("L")
 
     if randomize_seed:
         seed = secrets.randbelow(MAX_SEED)
 
     prompt = prompt + ".\n" + SYSTEM_PROMPT if prompt else SYSTEM_PROMPT
-    batch_size = 4
+    batch_size = 1
     results_images = pipe(
         prompt=[prompt] * batch_size,
         image=[image] * batch_size,