Update app.py

app.py

@@ -1,6 +1,8 @@
 import torch
 import gradio as gr
+import os
 from pathlib import Path
+from huggingface_hub import login
 from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, Settings
 from llama_index.core.memory import ChatMemoryBuffer
 from llama_index.llms.huggingface import HuggingFaceLLM
@@ -37,6 +39,8 @@ Current conversation:
 {query_str} [/INST]"""
 )
 
+login(token=os.environ["HF_TOKEN"]) 
+
 llm = HuggingFaceLLM(
     tokenizer_name="meta-llama/Llama-2-7b-chat-hf",
     model_name="meta-llama/Llama-2-7b-chat-hf",