Update app.py

app.py

@@ -6,6 +6,7 @@ from spacy_streamlit import visualize_ner
 import en_core_web_sm
 
 NER = spacy.load("en_core_web_sm")
+spacy_model = "en_core_web_sm"
 st.title('Entity Extraction')
 
 def prediction(raw_text):
@@ -13,6 +14,7 @@ def prediction(raw_text):
   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()
@@ -21,7 +23,7 @@ def prediction(raw_text):
   st.markdown(displacy.render(text1,style="ent"))
    
   models = ["en_core_web_sm"]
-  spacy_streamlit.visualize_spans(text1)
+  spacy_streamlit.visualize_ner(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.