Update handler.py

handler.py

@@ -8,7 +8,7 @@ import torch
 
 class EndpointHandler():
     def __init__(self, path=""):
-        self.pipeline = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16").to("cuda")
+        self.pipeline = AutoPipelineForInpainting.from_pretrained("OzzyGT/RealVisXL_V4.0_inpainting", torch_dtype=torch.float16, variant="fp16").to("cuda")
 
     def __call__(self, data: Dict[str, Any]):
         """
@@ -25,7 +25,9 @@ class EndpointHandler():
         image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
         mask = Image.open(BytesIO(base64.b64decode(inputs['mask'])))
         prompt = inputs['prompt']
-
+        # blur mask for smooth transition
+        mask = pipeline.mask_processor.blur(mask, blur_factor=20)
+        
         # fix the seed
         generator = torch.Generator(device="cuda").manual_seed(0)