Update handler.py

handler.py

@@ -32,7 +32,7 @@ class EndpointHandler():
 
 
         self.pipe = StableDiffusionControlNetInpaintPipeline.from_pretrained(
-            "diffusers/stable-diffusion-xl-1.0-inpainting-0.1", controlnet=self.controlnet, torch_dtype=torch.float16
+            "runwayml/stable-diffusion-v1-5", controlnet=self.controlnet, torch_dtype=torch.float16
         )
 
         self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(self.pipe.scheduler.config)
@@ -156,24 +156,21 @@ class EndpointHandler():
         # return first generate PIL image
         return image
         """
-        
-        control_image = self.make_inpaint_condition(image, mask_image)
 
+        control_image = self.make_inpaint_condition(image, mask_image)
 
         # generate image
         image = self.pipe(
             prompt=prompt,
-            image=image,
             negative_prompt=negative_prompt,
             num_inference_steps=num_inference_steps,
             eta=1.0,
+            image=image,
             mask_image=mask_image,
             control_image=control_image,
             guidance_scale=guidance_scale,
-            strength=strength,
+            strength=strength
         ).images[0]
-
-        return image
         
     
     # helper to decode input image