Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,6 +7,11 @@ QUESTION = "Compute the integral of f(x) = x^2."
 MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
 hf_token = os.getenv("HF_TOKEN")
 
+# Check if HF token is set
+if not hf_token:
+    st.error("HF_TOKEN is not set. Please add it to your HuggingFace secrets.")
+    st.stop()
+
 # remembers session
 if "response" not in st.session_state:
     st.session_state.response = ""
@@ -27,7 +32,8 @@ def get_llm(model_id=MODEL, max_new_tokens=130, temperature=0.7):
     llm = HuggingFaceEndpoint(
         repo_id=model_id,
         max_new_tokens=max_new_tokens,
-        temperature=temperature
+        temperature=temperature,
+        token=hf_token
     )
     return llm
     
@@ -82,6 +88,9 @@ with st.container():
                 if st.button("📝 Explain the question"):
                     prompt = f"[INST]You are a thoughtful AI assistant.\nUser: {QUESTION} [/INST]\nAI:"
                     st.session_state.response = llm.invoke(prompt)
+
+                    print("Model Response:", st.session_state.response)
+
                     
                     st.session_state.button_clicked = "Explain the question"
             with col2: