Update handler.py

handler.py

@@ -70,8 +70,8 @@ class EndpointHandler():
         
         # process image
         if encoded_image is not None and encoded_mask_image is not None:
-            image = self.decode_base64_image(encoded_image)
-            mask_image = self.decode_base64_image(encoded_mask_image)
+            image = self.decode_base64_image(encoded_image).convert("RGB")
+            mask_image = self.decode_base64_image(encoded_mask_image).convert("RGB")
         else:
             image = None
             mask_image = None 
@@ -96,7 +96,7 @@ class EndpointHandler():
         self.pipe.enable_xformers_memory_efficient_attention()
         
         # run inference pipeline
-        out = self.pipe(prompt=prompt, negative_prompt=negative_prompt, image=image, mask_image=mask_image, height=image.heigth, width=image.width)
+        out = self.pipe(prompt=prompt, negative_prompt=negative_prompt, image=image, mask_image=mask_image)
 
         print("1st pipeline part successful!")