Update app.py

app.py

@@ -45,6 +45,7 @@ db = create_database()
 #     st.session_state["model"] = AutoModelForSeq2SeqLM.from_pretrained(
 #         "MBZUAI/LaMini-Flan-T5-783M"
 #     )
+
 @st.cache_resource()
 def load_model():
     print("test")
@@ -93,8 +94,7 @@ if user_input:
     if tag in ["greeting"]:
         output = "Hello 👋! Thanks for visiting!\n I am BGPT! I am here to assist you in obtaining information about Bibek. Feel free to ask me any questions about Bibek. These are some sample questions:\n (I) Tell me about Bibek.\n (II) What skills does Bibek have?\n (III) What work experience does Bibek have?\n (IV) What is Bibek's educational background?\n (V) What awards has Bibek won?\n (VI) What projects have Bibek completed? &\n (VII) How can I contact Bibek?"
     else:
-        tokenizer = st.session_state["tokenizer"]
-        model = st.session_state["model"]
+        tokenizer, model = load_model()
         docs = db.similarity_search(user_input)
         output = wrap_text_preserve_newlines(str(docs[0]))
         if tag in ["welcome", "thanks", "exit"]: