Update handler.py

handler.py

@@ -21,10 +21,10 @@ class EndpointHandler:
         # move model to device
         self.model.to(device)
 
-    def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
+    def __call__(self, data: Any):
         logger.info(f"data: {data}")
         logger.info(f"data type: {type(data)}")
-        image = Image.open(BytesIO(base64.b64decode(data)))
+        image = Image.open(BytesIO(base64.b64decode(data["inputs"])))
         logger.info("image read")
         inputs = self.processor(image, return_tensors="pt")
         outputs = self.model(**inputs)