Upload app.py

app.py

@@ -21,6 +21,16 @@ def question_model():
     question_answerer = pipeline("question-answering", model=model, tokenizer=tokenizer)
     return question_answerer
 
+def question_answering(context, question):
+    with st.spinner(text="Loading question model..."):
+            question_answerer = question_model()
+        with st.spinner(text="Getting answer..."):
+            answer = question_answerer(context=context, question=question)
+            answer = answer["answer"]
+            # display the result in container
+            container = st.container(border=True)
+            container.write("<h5><b>Answer:</b></h5>" + answer + "<br>", unsafe_allow_html=True)
+
 # choose the source with different tabs
 tab1, tab2 = st.tabs(["Input text", "Upload File"])
 
@@ -48,16 +58,9 @@ with tab1:
     question = st.text_input(label="Enter the question: ", value=question, key="questionInput")
     
     # perform question answering when "get answer" button clicked
-    button = st.button("Get answer")
+    button = st.button("Get answer", key="textInput")
     if button:
-        with st.spinner(text="Loading question model..."):
-            question_answerer = question_model()
-        with st.spinner(text="Getting answer..."):
-            answer = question_answerer(context=context, question=question)
-            answer = answer["answer"]
-            # display the result in container
-            container = st.container(border=True)
-            container.write("<h5><b>Answer:</b></h5>" + answer + "<br>", unsafe_allow_html=True)
+        question_answering(context, question)
 
 
 # if upload file as input  
@@ -72,16 +75,9 @@ with tab2:
         question = st.text_input(label="Enter your question", value=sample_question)
 
         # perform question answering when "get answer" button clicked
-        button2 = st.button("Get answer")
+        button2 = st.button("Get answer", key="fileInput")
         if button2:
-            with st.spinner(text="Loading question model..."):
-                question_answerer = question_model()
-            with st.spinner(text="Getting answer..."):
-                answer = question_answerer(context=context, question=question)
-                answer = answer["answer"]
-                # display the result in container
-                container = st.container(border=True)
-                container.write("<h5><b>Answer:</b></h5>" + answer + "<br>", unsafe_allow_html=True)
+            question_answering(context, question)
 
 st.markdown("<br><br><br><br><br>", unsafe_allow_html=True)