Update app.py

app.py

@@ -16,8 +16,8 @@ def main():
     spacy_model = "en_core_web_sm"
     st.title('Entity Extraction')
 
-    menu = ["Home","NER"]
-    choice = st.sidebar.selectbox("Menu", menu)
+    #menu = ["Home","NER"]
+    #choice = st.sidebar.selectbox("Menu", menu)
     with st.sidebar:
         st.write("Sample Text")
         st.write("""Ai-Khanoum (/aɪ ˈhɑːnjuːm/, meaning Lady Moon; Uzbek: Oyxonim) is the archaeological site of a Hellenistic city in Takhar Province, Afghanistan. 
@@ -27,19 +27,17 @@ archaeologists until the outbreak of conflict in Afghanistan in the late 1970s.
         st.write("""The Indian Space Research Organisation or is the national space agency of India, headquartered in Bengaluru.
         It operates under Department of Space which is directly overseen by the Prime Minister of India while Chairman of ISRO acts as executive of DOS as well.""")
     
-    if choice == "Home":
-        st.subheader("Tokenization")
-        raw_text = st.text_area("Your Text","Enter the Text Here")
-        docx = nlp(raw_text)
-        if st.button("Tokenize"):
-            spacy_streamlit.visualize_tokens(docx,attrs = ['text','pos_','dep_','ent_type_'])
-
-    elif choice == 'NER':
-        st.subheader("Name Entity Recognition")
-        raw_text = st.text_area("Your Text","Enter the Text Here")
-        docx = nlp(raw_text)
+    st.subheader("Tokenization")
+    raw_text = st.text_area("Your Text","Enter the Text Here")
+    docx = nlp(raw_text)
+    if st.button("Tokenize"):
+        spacy_streamlit.visualize_tokens(docx,attrs = ['text','pos_','dep_','ent_type_'])
+
+    st.subheader("Name Entity Recognition")
+    if st.button("Entity Extraction"):
+           
         spacy_streamlit.visualize_ner(docx,labels = nlp.get_pipe('ner').labels,show_table = False)
-        #displacy.render(docx,style="ent")
+