Update app.py

app.py

@@ -120,20 +120,26 @@ def perform_inference(text,display_area):
     display_area.text("Initializing BERT module...")
     st.session_state['ner_phi'] = ner.UnsupNER("bbc/ner_bbc_config.json")
 
-
-
+  if (st.session_state['aggr'] is None):
+    display_area.text("Initializing Aggregation modeule...")
+    st.session_state['aggr'] = aggr.AggregateNER("./ensemble_config.json")
 
  
   display_area.text("Getting predictions from BERT model...")
   phi_results = st.session_state['phi_model'].get_descriptors(text,pos_arr)
   display_area.text("Computing NER results...")
 
+  display_area.text("Consolidating responses...")
   phi_ner = st.session_state['ner_phi'].tag_sentence_service(text,phi_results)
+  obj = phi_ner
+  combined_arr = [obj,obj]
+  aggregate_results = st.session_state['aggr'].fetch_all(text,combined_arr)
+  return aggregate_results
 
-  return phi_ner
 
 
 sent_arr = [
+"Washington who resigned from Washington flew to Washington",
 "John Doe flew from New York to Rio De Janiro ",
 "In 2020, John participated in the Winter Olympics and came third in Ice hockey",
 "Stanford called",
@@ -149,6 +155,7 @@ sent_arr = [
 
 
 sent_arr_masked = [
+"Washington:__entity__ who resigned from Washington:__entity__ flew to Washington:__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",