Update app.py

app.py

@@ -117,16 +117,16 @@ def perform_inference(text,display_area):
 
 
   if (st.session_state['ner_phi'] is None):
-    display_area.text("Initializing PHI module...")
+    display_area.text("Initializing BERT module...")
     st.session_state['ner_phi'] = ner.UnsupNER("bbc/ner_bbc_config.json")
 
 
 
 
  
-  display_area.text("Getting results from PHI model...")
+  display_area.text("Getting predictions from BERT model...")
   phi_results = st.session_state['phi_model'].get_descriptors(text,pos_arr)
-  display_area.text("Aggregating BIO & PHI results...")
+  display_area.text("Computing NER results...")
 
   phi_ner = st.session_state['ner_phi'].tag_sentence_service(text,phi_results)
 
@@ -143,12 +143,13 @@ sent_arr = [
 "I met my two friends at the pub",
 "The sky turned dark in advance of the storm that was coming from the east ",
 "She loves to watch Sunday afternoon football with her family ",
+"The United States has the largest prison population in the world, and the highest per-capita incarceration rate",
 "Paul Erdos died at 83 "
 ]
 
 
 sent_arr_masked = [
-"John:__entity__ Doe:__entity__ flew from New York to Rio:__entity__ De:__entity__ Janiro:__entity__ ",
+"John:__entity__ Doe:__entity__ flew from New:__entity__ York:__entity__ to Rio:__entity__ De:__entity__ Janiro:__entity__ ",
 "In 2020:__entity__, Catherine:__entity__ Zeta:__entity__ Jones:__entity__ participated in the Winter:__entity__ Olympics:__entity__ and came third in Ice:__entity__ hockey:__entity__",
 "Stanford:__entity__ called",
 "I met my girl:__entity__ friends at the pub ",
@@ -157,6 +158,7 @@ sent_arr_masked = [
 "I met my two:__entity__ friends at the pub",
 "The sky turned dark:__entity__ in advance of the storm that was coming from the east ",
 "She loves to watch Sunday afternoon football:__entity__ with her family ",
+"The United:__entity__ States:__entity__ has the largest prison population in the world, and the highest per-capita incarceration:__entity__ rate:__entity__",
 "Paul:__entity__ Erdos:__entity__ died at 83:__entity__ "
 ]