Update handler.py

handler.py

@@ -33,8 +33,9 @@ class EndpointHandler:
                 The payload with the text prompt and generation parameters.
         """
         # process input
-        image_base64 = data.pop("image_base64", None)
-        prompt = data.pop("prompt", None)
+        inputs = data.pop("inputs", data)
+        image_base64 = inputs["image_base64"]
+        prompt = inputs["prompt"]
 
         # preprocess
         image = Image.open(BytesIO(base64.b64decode(image_base64)))