Update app.py

app.py

@@ -8,20 +8,19 @@ st.title('Entity Extraction')
 
 def prediction(raw_text):
   text1= NER(raw_text)
-  print("List wise NERs:")
-  print("------------------")
-  print(f"{'Text' : <10}{'NER' : >10}")
+  st.write("List wise NERs:")
+  st.write("------------------")
+  st.write(f"{'Text' : <10}{'NER' : >10}")
   for word in text1.ents:
-    print(word.text,"\t\t",word.label_)
+    st.write(word.text,"\t\t",word.label_)
   print()
-  print("------------------")
-  print("NERs in the sentence:")
-  return displacy.render(text1,style="ent")
+  st.write("------------------")
+  st.write("NERs in the sentence:")
+  st.markdown(displacy.render(text1,style="ent"))
 
 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. """
-x = prediction(raw_text)
-html_string = "<h3>x</h3>"
-st.markdown(html_string,unsafe_allow_html=True)
+prediction(raw_text)
+