a

handler.py

@@ -70,7 +70,7 @@ class EndpointHandler():
     # helper to decode input image
     def decode_base64_image(self, image_string):
         base64_image = base64.b64decode(image_string)
-        buffer = io.BytesIO(base64_image)
+        buffer = BytesIO(base64_image)
         image = Image.open(buffer)
         pil_image = Image.fromarray(np.array(image))
         return pil_image