Update app.py

app.py

@@ -21,26 +21,29 @@ st.markdown("""
     </style>
     """, unsafe_allow_html=True)
 
-opt=st.sidebar.radio("Select a PubMed Database", options=('Clotting', 'Neuroblastoma'))
-if opt == "Clotting":
+opt=st.sidebar.radio("Select a PubMed Corpus", options=('Clotting corpus', 'Neuroblastoma corpus'))
+if opt == "Clotting corpus":
     model_used = ("pubmed_model_clotting")
+    num_abstracts = 45493
     database_name = "Clotting"
-if opt == "Neuroblastoma":
-    model_used = ("pubmed_model_clotting")
+if opt == "Neuroblastoma corpus":
+    model_used = ("pubmed_model_neuroblastoma")
+    num_abstracts = 29032
     database_name = "Neuroblastoma"
 
 
-st.header(f"Word2Vec App for {database_name} Pubmed Database.")
+st.header(f"Word2Vec App for {database_name} Pubmed corpus.")
 st.subheader("Uncovering knowledge through Natural Language Processing (NLP)")
+st.subheader("Open sidebar to choose corpus")
 
-text_input_value = st.text_input(f"Enter one term to search within the {database_name} database", max_chars=50)
+text_input_value = st.text_input(f"Enter one term to search within the {database_name} corpus", max_chars=50)
 query = text_input_value
 query = query.lower()
 # query = input ("Enter your keyword(s):")
 if query:
     bar = st.progress(0)
     time.sleep(.2)
-    st.caption(f":LightSkyBlue[searching 40123 {database_name} PubMed abstracts]")
+    st.caption(f":LightSkyBlue[searching {num_abstracts} {database_name} PubMed abstracts]")
     for i in range(10):
         bar.progress((i + 1) * 10)
         time.sleep(.1)