Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 st.title("📚 Study Buddy Chatbot")
@@ -23,12 +23,15 @@ def load_model():
     return tokenizer, model
 
 # Only load model when needed
-if "model_loaded" not in st.session_state:
+if "tokenizer" not in st.session_state or "model" not in st.session_state:
     with st.spinner("Loading AI model (this may take a minute)..."):
-        tokenizer, model = load_model()
-        st.session_state.model_loaded = True
+        st.session_state.tokenizer, st.session_state.model = load_model()
 
 def get_response(user_input):
+    # Get tokenizer and model from session state
+    tokenizer = st.session_state.tokenizer
+    model = st.session_state.model
+    
     # Format conversation history for context
     history = "\n".join(st.session_state.conversation[-6:])  # Last 6 exchanges