Update app.py

app.py

@@ -23,7 +23,7 @@ nlp = spacy.load("en_core_web_sm")
 user_agent = 'QGen/1.0 ([email protected])'
 wiki_wiki = wikipediaapi.Wikipedia(user_agent= user_agent,language='en')
 
-@st.cache_resources(allow_output_mutation=True)
+@st.cache_resource(allow_output_mutation=True)
 def load_model():
     model_name = "DevBM/t5-large-squad"
     model = T5ForConditionalGeneration.from_pretrained(model_name)
@@ -182,7 +182,7 @@ with st.sidebar:
     question_complexity = st.selectbox("Select question complexity", ["Simple", "Intermediate", "Complex"])
 if st.button("Generate Questions"):
     if text:
-        load_model()
+        model, tokenizer = load_model()
         keywords = extract_keywords(text)
         keyword_sentence_mapping = map_keywords_to_sentences(text, keywords, context_window_size)