Update app.py

app.py

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