Update app.py

app.py

@@ -139,7 +139,7 @@ def run_inpainting(image, replaced_caption, masks, generator, device):
     ).to(device)
 
     masks = Image.fromarray(masks.numpy())
-    dilation_image = masks.filter(ImageFilter.MaxFilter(3))
+    # dilation_image = masks.filter(ImageFilter.MaxFilter(3))
 
     prompt = replaced_caption
     negative_prompt = """lowres, bad anatomy, bad hands,
@@ -149,7 +149,8 @@ def run_inpainting(image, replaced_caption, masks, generator, device):
     output = pipeline(
         prompt=prompt,
         image=image,
-        mask_image=dilation_image,
+        # mask_image=dilation_image,
+        mask_image=masks,
         negative_prompt=negative_prompt,
         guidance_scale=7.5,
         strength=1.0,