Update app.py

app.py

@@ -19,9 +19,12 @@ from constants import CHROMA_SETTINGS
 st.set_page_config(layout="centered")
 
 checkpoint = "meta-llama/Llama-2-7b-chat-hf"
-tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+token = os.getenv("HF_TOKEN")
+
+tokenizer = AutoTokenizer.from_pretrained(checkpoint, use_auth_token=token)
 model = AutoModelForCausalLM.from_pretrained(
-    checkpoint,
+    checkpoint, 
+    use_auth_token=token,
     device_map="auto",
     torch_dtype=torch.float32
 )