Spaces:

abokbot
/

question-generation

Sleeping

abokbot commited on May 14, 2023

Commit

c62621e

•

1 Parent(s): b607935

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ st.header("Generate questions for short Wikipedia-like articles")
 st_model_load = st.text('Loading question generator model...')
-@st.cache(allow_output_mutation=True)
 def load_model():
     print("Loading model...")
     tokenizer = T5TokenizerFast.from_pretrained("t5-base")

 st_model_load = st.text('Loading question generator model...')
+@st.cache_resource
 def load_model():
     print("Loading model...")
     tokenizer = T5TokenizerFast.from_pretrained("t5-base")