fix: 添加hf_tocken

modules/ai_model.py

@@ -3,6 +3,8 @@ import torch
 import base64
 import requests
 from io import BytesIO
+import os
+from huggingface_hub import login
 from PIL import Image
 from transformers import AutoProcessor, Gemma3nForConditionalGeneration
 from utils.logger import log
@@ -15,10 +17,29 @@ class AIModel:
         self.processor = None
         self._initialize_model()
 
+    def _authenticate_hf(self):
+        def _authenticate_hf(self):
+        """HuggingFace认证"""
+        try:
+            # 从环境变量获取HF token
+            hf_token = os.getenv("Assitant_tocken") or os.getenv("Assitant_tocken")
+            
+            if hf_token:
+                login(token=hf_token, add_to_git_credential=False)
+                log.info("✅ HuggingFace 认证成功")
+            else:
+                log.warning("⚠️ 未找到 HuggingFace token，可能无法访问受限模型")
+                
+        except Exception as e:
+            log.error(f"❌ HuggingFace 认证失败: {e}")
+
+
     def _initialize_model(self):
         """初始化Gemma模型 - 基于官方调用方式"""
         try:
             log.info(f"正在加载模型: {self.model_name}")
+
+            hf_token = os.getenv("Assitant_tocken") or os.getenv("Assitant_tocken")
             
             self.model = Gemma3nForConditionalGeneration.from_pretrained(
                 self.model_name,
@@ -29,7 +50,8 @@ class AIModel:
             
             self.processor = AutoProcessor.from_pretrained(
                 self.model_name,
-                trust_remote_code=True
+                trust_remote_code=True,
+                tocken=hf_token
             )
             
             log.info("✅ Gemma AI 模型初始化成功")