chore: add Access Token

app.py

@@ -39,12 +39,12 @@ if cuda_path:
     print(f"CUDA 已安装在：{cuda_path}")
 else:
     print("未找到已安装的 CUDA 路径")
-    
+
 # 从 HF 加载预训练模型
 class LRMGeneratorWrapper:
     def __init__(self):
-        self.config = AutoConfig.from_pretrained("yanranxiaoxi/image-upscale", trust_remote_code=True)
-        self.model = AutoModel.from_pretrained("yanranxiaoxi/image-upscale", trust_remote_code=True)
+        self.config = AutoConfig.from_pretrained("yanranxiaoxi/image-upscale", trust_remote_code=True, token=os.environ.get('MODEL_ACCESS_TOKEN'))
+        self.model = AutoModel.from_pretrained("yanranxiaoxi/image-upscale", trust_remote_code=True, token=os.environ.get('MODEL_ACCESS_TOKEN'))
         self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         self.model.to(self.device)
         self.model.eval()