Allow to put context as well

app.py

@@ -8,7 +8,8 @@ onnx_qa = pipeline("question-answering", model=model, tokenizer=tokenizer)
 
 # question = "What's my name??"
 # context = "My name is Philipp and I live in Nuremberg."
-def get_answer(question):
+def get_answer(question, context):
+    # question, context = inputs
     pred = onnx_qa(question, context)['answer']
     return pred
 
@@ -21,7 +22,7 @@ with demo:
         question = gr.Textbox(label='Question', lines= 3)
         b1 = gr.Button('Get Answer')
         answer = gr.Textbox(label='Answer', lines=4)
-    b1.click(fn = get_answer, inputs=question, outputs=answer)
+    b1.click(fn = get_answer, inputs=[question,context], outputs=answer)
 
 demo.launch()