Update app.py

app.py

@@ -12,19 +12,24 @@ api_key = os.getenv("api_key")
 st.title("I am Your GrowBuddy 🌱")
 st.write("Let me help you start gardening. Let's grow together!")
 
-# Function to load model
+# Function to load model only once
 def load_model():
     try:
-        tokenizer = AutoTokenizer.from_pretrained("KhunPop/Gardening")
-        model = AutoModelForCausalLM.from_pretrained("unsloth/gemma-2-2b")
-        #tokenizer = AutoTokenizer.from_pretrained("KhunPop/Gardening", use_auth_token=api_key)
-        #model = AutoModelForCausalLM.from_pretrained("KhunPop/Gardening", use_auth_token=api_key)
-        return tokenizer, model
+        # If model and tokenizer are already in session state, return them
+        if "tokenizer" in st.session_state and "model" in st.session_state:
+            return st.session_state.tokenizer, st.session_state.model
+        else:
+            tokenizer = AutoTokenizer.from_pretrained("KhunPop/Gardening")
+            model = AutoModelForCausalLM.from_pretrained("unsloth/gemma-2-2b")
+            # Store the model and tokenizer in session state
+            st.session_state.tokenizer = tokenizer
+            st.session_state.model = model
+            return tokenizer, model
     except Exception as e:
         st.error(f"Failed to load model: {e}")
         return None, None
 
-# Load model and tokenizer
+# Load model and tokenizer (cached)
 tokenizer, model = load_model()
 
 if not tokenizer or not model: