Update app.py

app.py

@@ -10,9 +10,10 @@ import gradio as gr
 
 # 加載 LLaMA 模型
 from transformers import AutoTokenizer, AutoModelForCausalLM
+token = os.getenv("Git Access")
 
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf")
-model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-hf")
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf",use_auth_token=token)
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-hf",use_auth_token=token)
 
 # 定義推理函數
 def generate_text(prompt):