Allow to put context as well

app.py

@@ -6,10 +6,10 @@ tokenizer = AutoTokenizer.from_pretrained("deepset/roberta-base-squad2")
 
 onnx_qa = pipeline("question-answering", model=model, tokenizer=tokenizer)
 
-question = "What's my name??"
-context = "My name is Philipp and I live in Nuremberg."
+# question = "What's my name??"
+# context = "My name is Philipp and I live in Nuremberg."
 def get_answer(question):
-    pred = onnx_qa(question, context)
+    pred = onnx_qa(question, context)['answer']
     return pred
 
 
@@ -17,6 +17,7 @@ demo = gr.Blocks()
 
 with demo:
     with gr.Row():
+        context = gr.Textbox(label='Document', lines=10)
         question = gr.Textbox(label='Question', lines= 3)
         b1 = gr.Button('Get Answer')
         answer = gr.Textbox(label='Answer', lines=4)