Update handler.py

handler.py

@@ -1,4 +1,3 @@
-# +
 from typing import  Dict, List, Any
 from PIL import Image
 import base64
@@ -6,7 +5,6 @@ import torch
 import os
 from io import BytesIO
 from transformers import BlipForConditionalGeneration, BlipProcessor
-# -
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
@@ -32,7 +30,7 @@ class EndpointHandler():
         """
         inputs = data.pop("inputs", data)
         parameters = data.pop("parameters", {})
-
+ 
         raw_images = [Image.open(BytesIO(base64.b64decode(_img))) for _img in inputs]
                                      
         processed_images = self.processor(images=raw_images, return_tensors="pt")
@@ -43,4 +41,4 @@ class EndpointHandler():
             out = self.model.generate(**processed_images)
         captions = self.processor.batch_decode(out, skip_special_tokens=True)
 
-        return {"captions": captions}
+        return {"captions": captions}