bglearning commited on
Commit
b83ecd0
·
1 Parent(s): 3f8aea3

Use st.cache_resource instead of st.cache

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def set_text_input():
21
  def main():
22
  models = ["google/tapas-base", "deepset/tapas-large-nq-hn-reader"]
23
 
24
- @st.cache()
25
  def load_tokenizer():
26
  tokenizer = AutoTokenizer.from_pretrained(selected_model)
27
  return tokenizer
 
21
  def main():
22
  models = ["google/tapas-base", "deepset/tapas-large-nq-hn-reader"]
23
 
24
+ @st.cache_resource()
25
  def load_tokenizer():
26
  tokenizer = AutoTokenizer.from_pretrained(selected_model)
27
  return tokenizer