adust inputs

handler.py

@@ -33,11 +33,7 @@ class EndpointHandler():
         Return:
             A :obj:`dict`:. base64 encoded image
         """
-        prompt = data.pop("inputs", data)
-        url = data.pop("src", data)
-        print('image', url)
-        response = requests.get(url)
-        init_image = Image.open(BytesIO(response.content)).convert("RGB")
+        init_image = data.pop("inputs", data)
         init_image.thumbnail((512, 512))
 
         params = data.pop("parameters", data)
@@ -46,6 +42,7 @@ class EndpointHandler():
         num_inference_steps = params.pop("num_inference_steps", 25)
         guidance_scale = params.pop("guidance_scale", 7.5)
         negative_prompt = params.pop("negative_prompt", None)
+        prompt = params.pop("prompt", None)
         height = params.pop("height", None)
         width = params.pop("width", None)
         manual_seed = params.pop("manual_seed", -1)