Spaces:

flax-community
/

code-clippy-problem-solver

Runtime error

arampacha commited on Jul 22, 2021

Commit

eb175a0

1 Parent(s): 0a35d3e

remove caching tokenizer

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,6 @@ model_name = "flax-community/gpt-code-clippy-125M-apps-alldata"
 def get_model():
     return AutoModelForCausalLM.from_pretrained(model_name)
-@st.cache
 def get_tokenizer():
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     tokenizer.pad_token = tokenizer.eos_token

 def get_model():
     return AutoModelForCausalLM.from_pretrained(model_name)
 def get_tokenizer():
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     tokenizer.pad_token = tokenizer.eos_token