Update app.py

app.py

@@ -6,15 +6,15 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
 from peft import PeftModel, PeftConfig
 import io
 
+# Login using Hugging Face token stored in Space secrets
+login(token=os.getenv("HUGGINGFACEHUB_TOKEN"))
+
+
 st.set_page_config(page_title="AnthroBot", page_icon="🤖", layout="centered")
 
 # Load model & tokenizer
 @st.cache_resource
 from huggingface_hub import login
-
-# Log in using the token
-login(token=os.getenv("HUGGINGFACEHUB_TOKEN"))
-
 def load_model():
     peft_config = PeftConfig.from_pretrained("SallySims/AnthroBot_Model_Lora")
     base_model = AutoModelForCausalLM.from_pretrained(