Update handler.py

handler.py

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