Update app.py

app.py

@@ -18,7 +18,10 @@ def prediction(raw_text):
   st.write("------------------")
   st.write("NERs in the sentence:")
   st.markdown(displacy.render(text1,style="ent"))
-  visualize_ner(text1, labels=nlp.get_pipe("ner").labels)
+
+  models = "en_core_web_sm"
+  spacy_streamlit.visualize(models, text1)
+  #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. 
 The city, whose original name is unknown,[a] was probably founded by an early ruler of the Seleucid Empire and served as a military and economic centre for the rulers of the Greco-Bactrian Kingdom until its destruction c. 145 BC.