Updated app.py 3 - rollback

app.py

@@ -13,7 +13,7 @@ st.header("BART-based Automatic Headline Generator")
 
 st_model_load = st.text('Loading BART headline generator model...')
 
-@st.cache_data(allow_output_mutation=True)
+@st.cache(allow_output_mutation=True)
 def load_model():
     print("Loading model...")
     tokenizer = AutoTokenizer.from_pretrained(model_name)