Update app.py

app.py

@@ -13,17 +13,16 @@ st.title("Epidemiology Extraction Pipeline for Rare Diseases")
 st.subheader("National Center for Advancing Translational Sciences (NIH/NCATS)") 
 
 
-
 #### CHANGE SIDEBAR WIDTH ###
 st.markdown(
     """
     <style>
     [data-testid="stSidebar"][aria-expanded="true"] > div:first-child {
-        width: 300px;
+        width: 275px;
     }
     [data-testid="stSidebar"][aria-expanded="false"] > div:first-child {
-        width: 300px;
-        margin-left: -500px;
+        width: 275px;
+        margin-left: -400px;
     }
     </style>
     """,
@@ -37,11 +36,17 @@ filtering = st.sidebar.radio("What type of filtering would you like?",('Strict',
 
 extract_diseases = st.sidebar.checkbox("Extract Rare Diseases", value=False)
 
-with st.spinner('Loading Epidemiology Models and Dependencies...'):
-    classify_model_vars = classify_abs.init_classify_model()
-    NER_pipeline, entity_classes = extract_abs.init_NER_pipeline()
-    GARD_dict, max_length = extract_abs.load_GARD_diseases()
-st.success('All Models and Dependencies Loaded!')
+@st.cache(suppress_st_warning=True)
+def load_models():
+    global classify_model_vars, NER_pipeline, entity_classes, GARD_dict, max_length
+    
+    with st.spinner('Loading Epidemiology Models and Dependencies...'):
+        classify_model_vars = classify_abs.init_classify_model()
+        NER_pipeline, entity_classes = extract_abs.init_NER_pipeline()
+        GARD_dict, max_length = extract_abs.load_GARD_diseases()
+    st.success('All Models and Dependencies Loaded!')
+
+load_models()
 
 disease_or_gard_id = st.text_input("Input a rare disease term or GARD ID.", value="Fellman syndrome")