fix load image

handler.py

@@ -5,6 +5,7 @@ import base64
 from io import BytesIO
 from pathlib import Path
 import os
+from diffusers.utils import load_image
 
 class EndpointHandler():
     def __init__(self, path=""):
@@ -22,9 +23,12 @@ class EndpointHandler():
          Return:
             A :obj:`list` | `dict`: will be serialized and returned
         """
-        input_image= data.pop("input_image", data)
+        input_image_url = data.pop("input_image", data)
         prompt = data.pop("prompt", None)
 
+        #load image
+        input_image = load_image(input_image_url)
+        
         # run normal prediction
         output = self.pipeline(prompt=prompt, image=input_image)