Update handler.py

handler.py

@@ -1,6 +1,7 @@
 # +
 from typing import  Dict, List, Any
 from PIL import Image
+import base64
 import torch
 import os
 from io import BytesIO
@@ -34,16 +35,16 @@ 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)
-        processed_image = {**processed_image, **parameters}
+        processed_images = self.processor(images=raw_images, return_tensors="pt")
+        processed_images["pixel_values"] = processed_images["pixel_values"].to(device)
+        processed_images = {**processed_images, **parameters}
         
         with torch.no_grad():
             out = self.model.generate(
-                **processed_image
+                **processed_images
             )
         captions = self.processor.batch_decode(out, skip_special_tokens=True)
-        # postprocess the prediction
+
         return {"captions": captions}