Update app.py

app.py

@@ -1,12 +1,24 @@
 # -*- coding: utf-8 -*-
-"""Hugging Face Llama 2 App"""
+"""Hugging Face Space App with Authentication"""
 
+import os
 import gradio as gr
+from huggingface_hub import login
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# 加載 LLaMA-2 模型
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-13b-chat-hf")
-model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-13b-chat-hf")
+# 登錄 Hugging Face，使用訪問令牌進行身份驗證
+HF_TOKEN = os.getenv("HF_TOKEN")  # 從環境變數中獲取訪問令牌
+if not HF_TOKEN:
+    raise ValueError(
+        "未找到 Hugging Face 訪問令牌！請設置環境變數 'HF_TOKEN'，或者直接提供有效的訪問令牌。"
+    )
+
+login(HF_TOKEN)  # 使用訪問令牌進行身份驗證
+
+# 加載 Llama-2-13b-chat-hf 模型
+MODEL_NAME = "meta-llama/Llama-2-13b-chat-hf"
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_TOKEN)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, use_auth_token=HF_TOKEN)
 
 # 定義推理函數
 def generate_text(prompt):