Update app.py

app.py

@@ -6,12 +6,21 @@ import en_core_web_sm
 NER = spacy.load("en_core_web_sm")
 st.title('Entity Extraction')
 
-raw_text="The Indian Space Research Organisation or is the national space agency of India, headquartered in Bengaluru. It operates under Department of Space which is directly overseen by the Prime Minister of India while Chairman of ISRO acts as executive of DOS as well."
-text1= NER(raw_text)
+def prediction(raw_text):
+  text1= NER(raw_text)
+  print("List wise NERs:")
+  print("------------------")
+  print(f"{'Text' : <10}{'NER' : >10}")
+  for word in text1.ents:
+    print(word.text,"\t\t",word.label_)
+  print()
+  print("------------------")
+  print("NERs in the sentence:")
+  return displacy.render(text1,style="ent")
 
-for word in text1.ents:
-    raw_text = raw_text.replace(word.text,"<b style='color:orange'>" + word + "</b>")
-raw_text = raw_replace("\n","<br>")
-    
-st.markdown(raw_text)
+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. 
+Rediscovered in 1961, the ruins of the city were excavated by a French team of archaeologists until the outbreak of conflict in Afghanistan in the late 1970s. """
+
+st.markdown(prediction(raw_text))