Update app.py

app.py

@@ -44,16 +44,18 @@ context_1 = st.text_area("Context #1", text_1, height=128)
 context_2 = st.text_area("Context #2", text_2, height=128)
 context_3 = st.text_area("New Context", value="", height=128)
 
-selected_context = st.radio("Select Context", ("Context #1", "Context #2", "New Context"))
+context = st.radio("Select Context", ("Context #1", "Context #2", "New Context"))
 
-if selected_context == "Context #1":
+if context == "Context #1":
+    selected_context = context_1
     selected_question = st.radio("Select Question", question_list_1)
-elif selected_context == "Context #2":
+elif context == "Context #2":
+    selected_context = context_2
     selected_question = st.radio("Select Question", question_list_2)
-elif selected_context == "New Context":
+elif context == "New Context":
+    selected_context = context_3
     selected_question = st.text_area("New Question", value="", height=64)
 
-
 def setModel(model_checkpoint):
     model = AutoModelForQuestionAnswering.from_pretrained(model_checkpoint)
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)