improve mask

models.py

@@ -238,10 +238,11 @@ def make_inpainting(positive_prompt: str,
         List[Image.Image]: list of generated images
     """
     pipe = get_inpainting_pipeline()
+    mask_image_postproc = convolution(mask_image)
 
     flush()
     st.success(f"{pipe.queue_size} images in the queue, can take up to {(pipe.queue_size+1) * 10} seconds")
-    image_ = pipe(image=image,
+    generated_image = pipe(image=image,
                     mask_image=Image.fromarray((mask_image * 255).astype(np.uint8)),
                     prompt=positive_prompt,
                     negative_prompt=negative_prompt,
@@ -250,6 +251,8 @@ def make_inpainting(positive_prompt: str,
                     width=WIDTH,
                     **common_parameters
                     ).images[0]
+    generated_image = postprocess_image_masking(generated_image, image, mask_image_postproc)
+
     return image_