update handler

handler.py

@@ -1,4 +1,3 @@
-from typing import Dict, List, Any
 from transformers import LayoutLMForTokenClassification, LayoutLMv2Processor, PegasusForConditionalGeneration, AlbertTokenizer
 import torch
 
@@ -11,12 +10,8 @@ class EndpointHandler():
         self.model = PegasusForConditionalGeneration.from_pretrained(model_id).to(device)
         self.tokenizer = AlbertTokenizer.from_pretrained(model_id)
 
-    def __call__(self, data: Dict[str, bytes]) -> Dict[str, List[Any]]:
-        """
-        Args:
-            data (:obj:):
-                includes the deserialized image file as PIL.Image
-        """
+    def __call__(self, data):
+
         # process input
         wrongSentence = data.pop("inputs", data)