added error handling

handler.py

@@ -107,18 +107,25 @@ class EndpointHandler:
         self.model = BertForQuestionAnswering.from_pretrained(path).to(device)
         self.tokenizer = BertTokenizer.from_pretrained(path)
 
-    def __call__(
-        self, 
-        data: Dict[str, Any]
-    ):
+    # def __call__(self, data: Dict[str, Any]):
+    # def __call__(self, data: dict[str, Any]) -> dict[str, list[Any]]:
+    def __call__(self, data: dict[str, Any]):
         """
         Args:
             data (:obj:):
-                includes the deserialized image file as PIL.Image
+                includes the context and question
         """
         try:
-            question = data.pop("question", data)
-            context = data.pop("context", data)
+            if 'inputs' not in data:
+                raise ValueError('no inputs key in data')
+
+            i = data.pop("inputs", data)
+            question = i.pop("question", False)
+            context = i.pop("context", False)
+
+            if question is False and context is False:
+                raise ValueError(
+                    f'No question and/or context: question: {question} - context: {context}')
 
             input_ids = self.tokenizer.encode(question, context)
             # print('The input has a total of {:} tokens.'.format(len(input_ids)))