Update app.py

app.py

@@ -25,7 +25,7 @@ def qna():
 
 @st.cache(allow_output_mutation=True)
 def answergen(context, question):
-    """DistilBert"""   
+    """CoQUAD"""   
     tokenizer = DistilBertTokenizer.from_pretrained(model_path,return_token_type_ids = True)
     model = DistilBertForQuestionAnswering.from_pretrained(model_path, return_dict=False)
     encoding = tokenizer.encode_plus(question, context)
@@ -47,7 +47,7 @@ def news():
 
 #%%
 def main():
-    st.sidebar.header("COVID-19 Question Answering (CO-QA) system")
+    st.sidebar.header("COVID-19 Question Answering Dataset(CoQUAD)system")
     
     wallpaper = Image.open('covid-removebg-preview.png')
     wallpaper = wallpaper.resize((1400,700))      
@@ -61,7 +61,7 @@ def main():
     st.markdown("<div align='center'><br>"
                 "<img src='https://img.shields.io/badge/Domain-Medical-blue'"
                  "alt='API stability' height='25'/>"
-                 "<img src='https://img.shields.io/badge/Model-covid__qa__distillbert-red'"
+                 "<img src='https://img.shields.io/badge/Model-CoQUAD-red'"
                  "alt='API stability' height='25'/>"
                  "<img src='https://img.shields.io/badge/Web%20App-Streamlit-yellow'"
                  "alt='API stability' height='25'/></div>", unsafe_allow_html=True)