Update app.py

app.py

@@ -13,13 +13,13 @@ st.header("Generate summaries")
 st_model_load = st.text('Loading summary generator model...')
 
 # # @st.cache(allow_output_mutation=True)
-# @st.cache_data
-# def load_model():
-print("Loading model...")
-# tokenizer = AutoTokenizer.from_pretrained(model_name)
-# model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-# nltk.download('punkt')
-# print("Model loaded!")
+@st.cache_data
+def load_model():
+    print("Loading model...")
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+    nltk.download('punkt')
+    print("Model loaded!")
 
 # tokenizer, model = load_model()
 st.success('Model loaded!')
@@ -99,7 +99,7 @@ def generate_summary():
     st.session_state.summaries = predicted_summaries
 
 # generate title button
-# st_generate_button = st.button('Generate summary', on_click=generate_summary)
+st_generate_button = st.button('Generate summary', on_click=generate_summary)
 
 # title generation labels
 if 'summaries' not in st.session_state: