Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import torch
+import os
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel, PeftConfig
 from huggingface_hub import login
@@ -8,16 +9,30 @@ from huggingface_hub import login
 st.set_page_config(page_title="LLaMA Chatbot", page_icon="🦙")
 status_placeholder = st.empty()
 
-# Set your HuggingFace token
+# Try multiple ways to get the token
 try:
-    hf_token = st.secrets["HF_TOKEN"]
-    status_placeholder.success("🔑 Successfully found HF token in secrets!")
+    # Try environment variable first
+    hf_token = os.environ.get("HF_TOKEN")
+    
+    if not hf_token:
+        # Try Streamlit secrets
+        hf_token = st.secrets.get("HF_TOKEN")
+    
+    if not hf_token:
+        raise ValueError("Token not found in any location")
+        
+    status_placeholder.success("🔑 Successfully found HF token!")
     login(token=hf_token)
     status_placeholder.success("🔑 Successfully logged in to Hugging Face!")
+    
 except Exception as e:
     status_placeholder.error(f"🚫 Error with HF token: {str(e)}")
+    st.error("Please ensure the token is properly set in the Space settings")
+    st.info("Current search paths for token:")
+    st.code("""
+    1. Environment variable: HF_TOKEN
+    2. Streamlit secrets: secrets.toml
+    """)
     st.stop()
 
-st.title("🦙 LLaMA Chatbot")
-
-# Rest of the code...
+st.title("🦙 LLaMA Chatbot")