Update handler.py

handler.py

@@ -33,8 +33,10 @@ class EndpointHandler():
         """
         print(data)
         try:
-            context = data.pop("context",None)
-            question = data.pop("question",None)
+            inputs = data.pop("inputs",None)
+            context = inputs["context"]
+            question = inputs["question"]
+            
             messages = generate_rag_prompt_message(context, question)
     
             inputs = self.tokenizer(