Cropinky commited on
Commit
0d61d97
·
1 Parent(s): e752124

st cache blabla

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from streamlit.elements.altair import generate_chart
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
4
  from transformers import pipeline
5
 
6
- @st.cache(ignore_hash=True)
7
  def load_model():
8
  model_ckpt = "flax-community/gpt2-rap-lyric-generator"
9
  tokenizer = AutoTokenizer.from_pretrained(model_ckpt,from_flax=True)
 
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
4
  from transformers import pipeline
5
 
6
+ @st.cache(allow_output_mutation=True)
7
  def load_model():
8
  model_ckpt = "flax-community/gpt2-rap-lyric-generator"
9
  tokenizer = AutoTokenizer.from_pretrained(model_ckpt,from_flax=True)