Read inputs from base64

handler.py

@@ -1,6 +1,9 @@
 from typing import Dict, List, Any
 from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
 import torch
+import base64
+from PIL import Image
+import io
 
 class EndpointHandler():
     def __init__(self, path=""):
@@ -10,8 +13,11 @@ class EndpointHandler():
             self.pipeline.to("cuda")
 
     def __call__(self, data: Dict[str, any]) -> List[Dict[str, Any]]:
-        mask = data["mask"].convert('RGB')
-        image = data["image"].convert('RGB')
+        mask_bytes = base64.b64decode(data["inputs"]["mask"])
+        mask = Image.open(io.BytesIO(mask_bytes)).convert('RGB')
+
+        image_bytes = base64.b64decode(data["inputs"]["image"])
+        image = Image.open(io.BytesIO(image_bytes)).convert('RGB')
 
         prompt = "a picture of a person with a nice haircut"