Update app.py

app.py

@@ -1,5 +1,6 @@
 import nltk
 nltk.download('stopwords')
+nltk.download('punkt')
 import pandas as pd
 #classify_abs is a dependency for extract_abs
 import classify_abs
@@ -14,8 +15,8 @@ st.subheader("National Center for Advancing Translational Sciences (NIH/NCATS)")
 #### CHANGE SIDEBAR WIDTH ###
 st.markdown(f'''
     <style>
-        section[data-testid="stSidebar"] .css-ng1t4o {{width: 10rem;}}
-        section[data-testid="stSidebar"] .css-1d391kg {{width: 10rem;}}
+        section[data-testid="stSidebar"] .css-ng1t4o {{width: 6rem;}}
+        section[data-testid="stSidebar"] .css-1d391kg {{width: 6rem;}}
     </style>
 ''',unsafe_allow_html=True)
 
@@ -28,9 +29,7 @@ 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()
-    st.success('Epidemiology Classification Model Loaded!')
     NER_pipeline, entity_classes = extract_abs.init_NER_pipeline()
-    st.success('Epidemiology Extraction Model Loaded!')
     GARD_dict, max_length = extract_abs.load_GARD_diseases()
 st.success('All Models and Dependencies Loaded!')