arampacha commited on
Commit
eb175a0
1 Parent(s): 0a35d3e

remove caching tokenizer

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -8,7 +8,6 @@ model_name = "flax-community/gpt-code-clippy-125M-apps-alldata"
8
  def get_model():
9
  return AutoModelForCausalLM.from_pretrained(model_name)
10
 
11
- @st.cache
12
  def get_tokenizer():
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
  tokenizer.pad_token = tokenizer.eos_token
 
8
  def get_model():
9
  return AutoModelForCausalLM.from_pretrained(model_name)
10
 
 
11
  def get_tokenizer():
12
  tokenizer = AutoTokenizer.from_pretrained(model_name)
13
  tokenizer.pad_token = tokenizer.eos_token