Upload app.py

app.py

@@ -15,7 +15,7 @@ st.write("Extractive question answering is a Natural Language Processing task wh
 @st.cache_resource(show_spinner=True)
 def question_model():
     # call my model for question answering
-    model_name = "kxx-kkk/FYP_deberta-v3-base-squad2_squad_adversarial"
+    model_name = "kxx-kkk/FYP_deberta-v3-base-squad2_mrqa"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForQuestionAnswering.from_pretrained(model_name)
     question_answerer = pipeline("question-answering", model=model, tokenizer=tokenizer)
@@ -36,7 +36,7 @@ def question_answering(context, question):
 # choose the source with different tabs
 tab1, tab2 = st.tabs(["Input text", "Upload File"])
 
-'''--- input text ---'''
+'''----- input text -----'''
 # if type the text as input
 with tab1:  
     # set the example  
@@ -68,11 +68,11 @@ with tab1:
         else:
             question_answering(context, question)
 
-'''--- upload file ---'''
+'''----- upload file -----'''
 # if upload file as input  
 with tab2:
     # provide upload place
-    uploaded_file = st.file_uploader("Choose a .pdf file to upload", type=["pdf"])
+    uploaded_file = st.file_uploader("Choose a .txt file to upload", type=["txt"])
 
     # transfer file to context and allow ask question, then perform question answering 
     if uploaded_file is not None:
@@ -90,3 +90,4 @@ with tab2:
 
 st.markdown("<br><br><br><br><br>", unsafe_allow_html=True)   
 
+