Update handler.py

handler.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, List
 
-import torch
+import torch, re
 
 import transformers
 
@@ -36,7 +36,9 @@ class EndpointHandler:
      
       
     def __call__(self, data: Dict[dict, Any]) -> Dict[str, Any]:
-        question = data.pop("question", data)
+        inputs = data.pop("inputs", data)
+        
+        question = data.pop("question", None)
         
         context = data.pop("context", None)