Update app.py

app.py

@@ -1,18 +1,23 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+import os
+
+# 获取 Hugging Face 访问令牌
+hf_token = os.getenv("HF_API_TOKEN")
 
 # 定义模型名称（替换为您上传的模型名称）
 model_name = "larry1129/WooWoof_AI"  # 替换为您的模型名称
 
 # 加载分词器
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
 
 # 加载模型
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
     device_map="auto",
     torch_dtype=torch.float16,
+    use_auth_token=hf_token,
     trust_remote_code=True  # 如果你的模型使用自定义代码，请保留此参数
 )