app.py

app.py

@@ -88,8 +88,9 @@ def initialize_wac_gec():
             
             # 优先使用本地模型，如果不存在则自动下载
             wac_corrector = WhitespaceCorrector.from_pretrained(
-                model="./models/eo_larger_byte",
-                device=device
+                model="eo_larger_byte",
+                device=device,
+                download_dir="./models"
             )
             print(f"✅ WAC-GEC模型已加载 (设备: {device})")
             return True
@@ -105,7 +106,7 @@ def call_wac_gec(text):
         raise ValueError("⚠️ WAC-GEC模型未安装或加载失败")
     
     try:
-        corrected = wac_corrector.correct_text("texttexttexttexttext")
+        corrected = wac_corrector.correct_text(text)
         # 格式化输出以匹配DeepSeek的格式
         return f"[output]: {corrected}"
     except Exception as e: