Add HF token for private model access

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+import os
 
 # 預先定義 Hugging Face 模型
 MODEL_NAMES = {
@@ -8,11 +9,15 @@ MODEL_NAMES = {
     "DeepSeek-R1": "deepseek-ai/DeepSeek-R1",
 }
 
+HF_TOKEN = os.getenv("HF_TOKEN")
+
 
 def load_model(model_path):
-    tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
+    tokenizer = AutoTokenizer.from_pretrained(
+        model_path, trust_remote_code=True, token=HF_TOKEN
+    )
     model = AutoModelForCausalLM.from_pretrained(
-        model_path, torch_dtype=torch.float16, trust_remote_code=True
+        model_path, torch_dtype=torch.float16, trust_remote_code=True, token=HF_TOKEN
     ).cuda()
     return model, tokenizer