Update handler.py

handler.py

@@ -25,11 +25,12 @@ class EndpointHandler():
         """
 
 
-        inputs = data.pop("inputs", data)
-
+        image_data = data.pop("inputs", data)
         # decode base64 image to PIL
-        image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
-        prompt = inputs['prompt']
+        image = Image.open(BytesIO(base64.b64decode(image_data)))
+
+        parameters = data.pop("parameters", data)
+        prompt = parameters['prompt']
 
         images = self.pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
         return images[0]