Update handler.py

handler.py

@@ -1,4 +1,4 @@
-from base64 import b64encode
+from base64 import b64encode, b64decode
 from io import BytesIO
 from pathlib import Path
 
@@ -31,8 +31,10 @@ class EndpointHandler:
         image = data.pop("inputs", data)
 
         # This lets us pass local images as well while developing
-        if isinstance(image, bytes):
-            image = Image.open(BytesIO(image)) # .convert("RGB")
+        if isinstance(image, str):
+            image = Image.open(BytesIO(b64decode(image)))
+        elif isinstance(image, bytes):
+            image = Image.open(BytesIO(image))
 
         image = np.array(image)
         image = image[:, :, ::-1]  # RGB -> BGR