InsafQ commited on
Commit
f9bc743
·
verified ·
1 Parent(s): 09d32fe

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -14,7 +14,7 @@ st.header("Generate candidate titles for articles")
14
 
15
  st_model_load = st.text('Loading title generator model...')
16
 
17
- @st.cache_data()
18
  def load_model():
19
  print("Loading model...")
20
  tokenizer = T5Tokenizer.from_pretrained(model_name, trust_remote_code=True)
 
14
 
15
  st_model_load = st.text('Loading title generator model...')
16
 
17
+ @st.cache_resource
18
  def load_model():
19
  print("Loading model...")
20
  tokenizer = T5Tokenizer.from_pretrained(model_name, trust_remote_code=True)