Update config.py

config.py

@@ -1,42 +1,42 @@
-import os
-from dataclasses import dataclass
-
-
-@dataclass
-class Config:
-    api_key: str
-    model_path: str
-    # 视觉语言模型 - 本地
-    vl_model_path: str
-    # 视觉语言模型 - 在线 API (DashScope)
-    vl_api_base: str
-    vl_api_key: str
-    vl_api_model: str
-    vl_use_api: bool
-    # 在线 API 最大调用次数限制（防止被刷爆，超出后自动降级到本地模型）
-    vl_api_max_calls: int
-    # 无论是否使用在线 API，始终加载本地 Qwen3-VL-2B-Instruct 模型
-    vl_always_load_local: bool
-    # 服务
-    host: str
-    port: int
-    gradio_port: int
-    device: str
-
-
-def load_config() -> Config:
-    return Config(
-        api_key=os.getenv("XGUARD_API_KEY", "your-api-key"),
-        model_path=os.getenv("XGUARD_MODEL_PATH", "Alibaba-AAIG/YuFeng-XGuard-Reason-0.6B"),
-        vl_model_path=os.getenv("XGUARD_VL_MODEL_PATH",""),
-        vl_api_base=os.getenv("XGUARD_VL_API_BASE", "https://dashscope.aliyuncs.com/compatible-mode/v1"),
-        vl_api_key=os.getenv("XGUARD_VL_API_KEY", ""),
-        vl_api_model=os.getenv("XGUARD_VL_API_MODEL", "qwen-vl-max-latest"),
-        vl_use_api=os.getenv("XGUARD_VL_USE_API", "").lower() in ("true", "1", "yes"),
-        vl_api_max_calls=int(os.getenv("XGUARD_VL_API_MAX_CALLS", "")),
-        vl_always_load_local=os.getenv("XGUARD_VL_ALWAYS_LOAD_LOCAL", "true").lower() in ("true", "1", "yes"),
-        host=os.getenv("XGUARD_HOST", "0.0.0.0"),
-        port=int(os.getenv("XGUARD_PORT", "8080")),
-        gradio_port=int(os.getenv("XGUARD_GRADIO_PORT", "7860")),
-        device=os.getenv("XGUARD_DEVICE", "auto"),
-    )
+import os
+from dataclasses import dataclass
+
+
+@dataclass
+class Config:
+    api_key: str
+    model_path: str
+    # 视觉语言模型 - 本地
+    vl_model_path: str
+    # 视觉语言模型 - 在线 API (DashScope)
+    vl_api_base: str
+    vl_api_key: str
+    vl_api_model: str
+    vl_use_api: bool
+    # 在线 API 最大调用次数限制（防止被刷爆，超出后自动降级到本地模型）
+    vl_api_max_calls: int
+    # 无论是否使用在线 API，始终加载本地 Qwen3-VL-2B-Instruct 模型
+    vl_always_load_local: bool
+    # 服务
+    host: str
+    port: int
+    gradio_port: int
+    device: str
+
+
+def load_config() -> Config:
+    return Config(
+        api_key=os.getenv("XGUARD_API_KEY", "your-api-key"),
+        model_path=os.getenv("XGUARD_MODEL_PATH", "Alibaba-AAIG/YuFeng-XGuard-Reason-0.6B"),
+        vl_model_path=os.getenv("XGUARD_VL_MODEL_PATH","Qwen/Qwen3-VL-2B-Instruct"),
+        vl_api_base=os.getenv("XGUARD_VL_API_BASE", "https://dashscope.aliyuncs.com/compatible-mode/v1"),
+        vl_api_key=os.getenv("XGUARD_VL_API_KEY", ""),
+        vl_api_model=os.getenv("XGUARD_VL_API_MODEL", "qwen-vl-max-latest"),
+        vl_use_api=os.getenv("XGUARD_VL_USE_API", "false").lower() in ("true", "1", "yes"),
+        vl_api_max_calls=int(os.getenv("XGUARD_VL_API_MAX_CALLS", "")),
+        vl_always_load_local=os.getenv("XGUARD_VL_ALWAYS_LOAD_LOCAL", "true").lower() in ("true", "1", "yes"),
+        host=os.getenv("XGUARD_HOST", "0.0.0.0"),
+        port=int(os.getenv("XGUARD_PORT", "8080")),
+        gradio_port=int(os.getenv("XGUARD_GRADIO_PORT", "7860")),
+        device=os.getenv("XGUARD_DEVICE", "auto"),
+    )