Update app.py

app.py

@@ -100,7 +100,7 @@ st.markdown("---")
 # if authenticate(password):
 opt = st.sidebar.radio("Select a PubMed Corpus", options=('Breast Cancer corpus', 'Lung Cancer corpus',
                                                           'Skin Cancer corpus', 'Colorectal Cancer corpus',
-                                                          'Prostate Cancer corpus', 'Lymphoma Cancer corpus'))
+                                                          'Prostate Cancer corpus', 'Lymphoma Cancer corpus', 'Urinary Cancer corpus'))
 # if opt == "Clotting corpus":
 #     model_used = ("pubmed_model_clotting")
 #     num_abstracts = 45493
@@ -133,6 +133,10 @@ if opt == "Lymphoma Cancer corpus":
     model_used = ("lymphoma_cancer_pubmed_model")
     num_abstracts = 79000
     database_name = "Lymphoma_cancer"
+if opt == "Urinary Cancer corpus":
+    model_used = ("urinary_cancer_pubmed_model")
+    num_abstracts = 65000
+    database_name = "Urinary_cancer"
 
 st.header(f":blue[{database_name} Pubmed corpus.]")
 text_input_value = st.text_input(f"Enter one term to search within the {database_name} corpus")