Update handler.py

handler.py

@@ -25,7 +25,8 @@ class EndpointHandler:
     
     def __call__(self, data: Dict[str, Any]) -> str:
         # Prepare your messages with image and text
-        base64image = data.get("base64")
+        inputs = data.get("inputs")
+        base64image = inputs["base64"]
 
         img_bytes = base64.b64decode(base64image)
         pil_img = Image.open(io.BytesIO(img_bytes)).convert("RGB")