Update app.py

app.py

@@ -11,19 +11,19 @@ st.title('Entity Extraction')
 
 def prediction(raw_text):
   text1= NER(raw_text)
-  #st.write("List wise NERs:")
-  #st.write("------------------")
-  #st.write(f"{'Text' : <10}{'NER' : >10}")
+  st.write("List wise NERs:")
+  st.write("------------------")
+  st.write(f"{'Text' : <10}{'NER' : >10}")
   
   for word in text1.ents:
     st.write(word.text,"\t\t",word.label_)
   print()
   st.write("------------------")
-  #st.write("NERs in the sentence:")
-  #st.markdown(displacy.render(text1,style="ent"))
+  st.write("NERs in the sentence:")
+  spacy_streamlit.visualize(displacy.render(text1,style="ent"))
    
   models = ["en_core_web_sm"]
-  spacy_streamlit.visualize_ner(text1,labels = word.label_)
+  #spacy_streamlit.visualize_ner(text1,labels = word.label_)
   #visualize_ner(text1, labels=nlp.get_pipe("ner").labels)
 
 raw_text = """Ai-Khanoum (/aɪ ˈhɑːnjuːm/, meaning Lady Moon; Uzbek: Oyxonim) is the archaeological site of a Hellenistic city in Takhar Province, Afghanistan.