Update app.py

app.py

@@ -1,8 +1,9 @@
 import spaces
 import gradio as gr
 from transformers import AutoTokenizer, LlamaForCausalLM
+import os
 
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf", use_auth_token=os.getenv("HUGGINGFACE_TOKEN"))
 model = LlamaForCausalLM.from_pretrained("kaist-ai/Prometheus-13b-v1.0", device_map="auto", load_in_8bit=True)
 
 @spaces.GPU