Update app.py

app.py

@@ -87,6 +87,8 @@ def init_session_states():
     st.session_state['bio_model'] = None
   if 'ner_bio' not in st.session_state:
     st.session_state['ner_bio'] = None
+  if 'aggr' not in st.session_state:
+    st.session_state['aggr'] = None
   
 
     
@@ -119,6 +121,10 @@ def perform_inference(text,display_area):
     display_area.text("Initializing BIO module...")
     st.session_state['ner_bio'] = ner.UnsupNER("bio/ner_a100_config.json")
     
+  if (st.session_state['aggr'] is None):
+    display_area.text("Initializing Aggregation module...")
+    st.session_state['aggr'] = aggr.AggregateNER("./ensemble_config.json")
+    
   
   
   display_area.text("Getting predictions from BIO model...")
@@ -126,12 +132,16 @@ def perform_inference(text,display_area):
   display_area.text("Computing BIO results...")
   bio_ner = st.session_state['ner_bio'].tag_sentence_service(text,bio_descs)
   
-  return bio_ner
+  obj = json.loads(bio_ner)
+  combined_arr = [obj,obj]
+  aggregate_results = st.session_state['aggr'].fetch_all(text,combined_arr)
+  return aggregate_results
   
 
 sent_arr = [
 "Lou Gehrig who works for XCorp and lives in New York suffers from Parkinson's ",  
 "Parkinson who works for XCorp and lives in New York suffers from Lou Gehrig's", 
+"Her hypophysitis secondary to ipilimumab was well managed with supplemental hormones",
 "lou gehrig was diagnosed with Parkinson's ",  
 "A eGFR below 60 indicates chronic kidney disease", 
 "Overexpression of EGFR occurs across a wide range of different cancers", 
@@ -147,6 +157,7 @@ sent_arr = [
 sent_arr_masked = [
 "Lou Gehrig:__entity__ who works for XCorp:__entity__ and lives in New:__entity__ York:__entity__ suffers from Parkinson's:__entity__ ",  
 "Parkinson:__entity__ who works for XCorp:__entity__ and lives in New:__entity__ York:__entity__ suffers from Lou Gehrig's:__entity__", 
+"Her hypophysitis:__entity__ secondary to ipilimumab:__entity__ was well managed with supplemental:__entity__ hormones:__entity__",
 "lou:__entity__ gehrig:__entity__ was diagnosed with Parkinson's:__entity__ ",  
 "A eGFR:__entity__ below 60 indicates chronic kidney disease", 
 "Overexpression of EGFR:__entity__ occurs across a wide range of different cancers",