Update edit_space.py

edit_space.py

@@ -303,7 +303,7 @@ class KontextEditModel():
 
         original_image_tensor = image.clone()
         original_mask = remove_mask
-        original_mask = self._expand_mask(remove_mask, expand=25)
+        original_mask = self._expand_mask(remove_mask, expand=10)
         
         image_pil = self._tensor_to_pil(image)
         # image_pil.save("image_pil.png")
@@ -340,7 +340,7 @@ class KontextEditModel():
                    seed, steps, cfg):
         generator = torch.Generator(device=self.device).manual_seed(seed)
         original_image_tensor = image.clone()
-        original_mask = self._expand_mask(fill_mask, expand=25)
+        original_mask = self._expand_mask(fill_mask, expand=10)
         image_pil = self._tensor_to_pil(image)
         # image_pil.save("image_pil.png")