Update app.py

app.py

@@ -20,10 +20,10 @@ def prediction(raw_text):
   print()
   st.write("------------------")
   st.write("NERs in the sentence:")
-  spacy_streamlit.visualize(displacy.render(text1,style="ent"))
+  #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 = text1.ent)
   #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.