DevBM commited on
Commit
c882126
1 Parent(s): e60bf9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ nlp = spacy.load("en_core_web_sm")
23
  user_agent = 'QGen/1.0 (channingfisher7@gmail.com)'
24
  wiki_wiki = wikipediaapi.Wikipedia(user_agent= user_agent,language='en')
25
 
26
- @st.cache(allow_output_mutation=True)
27
  def load_model():
28
  model_name = "DevBM/t5-large-squad"
29
  model = T5ForConditionalGeneration.from_pretrained(model_name)
 
23
  user_agent = 'QGen/1.0 (channingfisher7@gmail.com)'
24
  wiki_wiki = wikipediaapi.Wikipedia(user_agent= user_agent,language='en')
25
 
26
+ @st.cache_resources(allow_output_mutation=True)
27
  def load_model():
28
  model_name = "DevBM/t5-large-squad"
29
  model = T5ForConditionalGeneration.from_pretrained(model_name)