Update app.py

app.py

@@ -1,12 +1,18 @@
 import streamlit as st
 # Load model directly
 from transformers import AutoModel
+from huggingface_hub import login
+import os
 
 file = 'llama-2-7b.Q4_K_M.gguf'
 NO_GPU = 0
 GPU_LAYERS = 50
 model = AutoModel.from_pretrained("TheBloke/Llama-2-7B-GGUF" + file, 
                                   model_file=file, model_type="llama", gpu_layers=NO_GPU)
+
+access_token = os.getenv('HF_TOKEN')
+login(token = access_token)
+
 prompt = "AI is going to"
 
 with st.container():