Update handler.py

handler.py

@@ -32,7 +32,7 @@ class EndpointHandler():
 
 
         self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
-            "runwayml/stable-diffusion-v1-5", controlnet=self.controlnet, torch_dtype=torch.float16
+            "diffusers/stable-diffusion-xl-1.0-inpainting-0.1", controlnet=self.controlnet, torch_dtype=torch.float16
         )
 
         self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(self.pipe.scheduler.config)
@@ -171,6 +171,8 @@ class EndpointHandler():
             guidance_scale=guidance_scale,
             strength=strength
         ).images[0]
+
+        return image
         
     
     # helper to decode input image