Update handler.py

handler.py

@@ -33,7 +33,7 @@ class EndpointHandler():
         prompt = data.pop("prompt", "")
 
         if encoded_image is not None:
-            image = self.decode_base64_image(encoded_image)
+            image = self.decode_base64_image(encoded_image).convert('RGB')
 
             image_processor = VaeImageProcessor();
             latents = image_processor.preprocess(image)