Update app.py

app.py

@@ -38,13 +38,23 @@ def create_database():
 
 db = create_database()
 
-if "tokenizer" not in st.session_state:
-    st.session_state["tokenizer"] = AutoTokenizer.from_pretrained(
-        "MBZUAI/LaMini-Flan-T5-783M"
+# if "tokenizer" not in st.session_state:
+#     st.session_state["tokenizer"] = AutoTokenizer.from_pretrained(
+#         "MBZUAI/LaMini-Flan-T5-783M"
+#     )
+#     st.session_state["model"] = AutoModelForSeq2SeqLM.from_pretrained(
+#         "MBZUAI/LaMini-Flan-T5-783M"
+#     )
+@st.cache_resource()
+def load_model():
+    print("test")
+    tokenizer = AutoTokenizer.from_pretrained(
+        "./generative_model/LaMini-Flan-T5-783M"
     )
-    st.session_state["model"] = AutoModelForSeq2SeqLM.from_pretrained(
-        "MBZUAI/LaMini-Flan-T5-783M"
+    model = AutoModelForSeq2SeqLM.from_pretrained(
+        "./generative_model/LaMini-Flan-T5-783M"
     )
+    return tokenizer, model
 
 st.title("BGPT : Bibek's Personal Chatbot")
 # Storing the chat