Fixed base64 decode

handler.py

@@ -4,6 +4,7 @@ from PIL import Image
 import torch
 import os
 from io import BytesIO
+import base64
 from transformers import BlipForConditionalGeneration, BlipProcessor
 # -
 
@@ -34,7 +35,7 @@ class EndpointHandler():
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", {})
  
-        raw_images = [Image.open(BytesIO(_img)) for _img in inputs]
+        raw_images = [Image.open(BytesIO(base64.b64decode(_img))) for _img in inputs]
                                      
         processed_image = self.processor(images=raw_images, return_tensors="pt") 
         processed_image["pixel_values"] = processed_image["pixel_values"].to(device)