Update handler.py

handler.py

@@ -16,12 +16,15 @@ class EndpointHandler():
 
         info=data['inputs']
         image=info.pop("image",data)
-        prompt=info.pop("prompt",data)
+        prompt=info.pop("text",data)
         image=base64.b64decode(image)
         raw_images = Image.open(BytesIO(image)).convert('RGB')
         
         images = self.pipe(prompt, image=raw_images, num_inference_steps=25, image_guidance_scale=1).images
-        return json.dumps({'image':images[0]})
+        img=images[0]
+        img_data=img.tobytes()
+        img_base64=base64.b64encode(img_data).decode('utf-8')
+        return {'image':img_base64}