Update handler.py

handler.py

@@ -41,6 +41,8 @@ class EndpointHandler():
         encoded_mask_image = data.pop("mask_image", None)
         
         prompt = data.pop("prompt", "")
+
+        negative_prompt = data.pop("negative_prompt", "")
         
         # process image
         if encoded_image is not None and encoded_mask_image is not None:
@@ -53,7 +55,7 @@ class EndpointHandler():
         self.pipe.enable_xformers_memory_efficient_attention()
         
         # run inference pipeline
-        out = self.pipe(prompt=prompt, image=image, mask_image=mask_image)
+        out = self.pipe(prompt=prompt, negative_prompt=negative_prompt, image=image, mask_image=mask_image)
 
         image = out.images[0].resize((1024, 1024))
 
@@ -61,6 +63,7 @@ class EndpointHandler():
 
         image = self.pipe2(
             prompt=prompt,
+            negative_prompt=negative_prompt,
             image=image,
             mask_image=mask_image,
             guidance_scale=8.0,