Update app.py

app.py

@@ -23,7 +23,7 @@ def prediction(raw_text):
   #st.markdown(displacy.render(text1,style="ent"))
    
   models = ["en_core_web_sm"]
-  spacy_streamlit.visualize_ner(text1)
+  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.