update handler

handler.py

@@ -21,8 +21,9 @@ class EndpointHandler:
         self.PROCESSOR = LayoutLMv3Processor(self.FEATURE_EXTRACTOR, self.TOKENIZER)
         self.MODEL = LayoutLMv3ForSequenceClassification.from_pretrained("OtraBoi/document_classifier_testing").to(device)
 
-    def __call__(self, data: Dict):
-        image = Image.open(io.BytesIO(data["inputs"])).convert("RGB")
+    def __call__(self, data: Dict[str, bytes]):
+        #image = Image.open(io.BytesIO(data["inputs"])).convert("RGB")
+        image = data.pop("inputs", data)
         encoding = self.PROCESSOR(image, return_tensors="pt", padding="max_length", truncation=True)
         
         for k,v in encoding.items():