added more params

handler.py

@@ -20,16 +20,29 @@ class EndpointHandler():
             inputs (:obj:`string`)
             parameters (:obj:)
       Return:
-            A :obj:`string`:. Base64 encoded image string
+            A :obj:`string`:. image string
         """
 
 
-        image_data = data.pop("inputs", data)
+        image_data = data.pop('inputs', data)
         # decode base64 image to PIL
         image = Image.open(BytesIO(base64.b64decode(image_data)))
 
-        parameters = data.pop("parameters", data)
-        prompt = parameters['prompt']
+        parameters = data.pop('parameters', [])
+        prompt = parameters.pop('prompt', None)
+        negative_prompt = parameters.pop('negative_prompt', None)
+        num_inference_steps = parameters.pop('num_inference_steps', 10)
+        image_guidance_scale = parameters.pop('image_guidance_scale', 1.5)
+        guidance_scale = parameters.pop('guidance_scale', 7.5)
+
+
+        images = self.pipe(
+            prompt,
+            image = image,
+            negative_prompt = negative_prompt,
+            num_inference_steps = num_inference_steps,
+            image_guidance_scale = image_guidance_scale,
+            guidance_scale = guidance_scale
+        ).images
 
-        images = self.pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
         return images[0]