Update handler.py

handler.py

@@ -42,8 +42,8 @@ class EndpointHandler:
                 - "time": the time run predict
         """
         inputs = data.pop("inputs", data)
-        context = inputs.context
-        question = inputs.question
+        context = inputs.pop("context", inputs)
+        question = inputs.pop("question", inputs)
 
         prompt = f"""Answer the question based on the context below. If the question cannot be answered using the information provided answer with 'No answer'. Stop response if end.
 >>TITLE<<: Flawless answer.