Просмотр исходного кода

fix: 修复路径导入问题,确保项目根目录被正确添加到系统路径

zhch158_admin 4 дней назад
Родитель
Сommit
dce163b944
1 измененных файлов с 6 добавлено и 0 удалено
  1. 6 0
      ocr_tools/universal_doc_parser/core/model_factory.py

+ 6 - 0
ocr_tools/universal_doc_parser/core/model_factory.py

@@ -1,5 +1,11 @@
 """模型工厂 - 根据配置创建模型实例"""
 from typing import Dict, Any, Optional
+import sys
+from pathlib import Path
+# 添加项目根目录到路径
+project_root = Path(__file__).parents[1]
+if str(project_root) not in sys.path:
+    sys.path.insert(0, str(project_root))
 from models.adapters import BasePreprocessor, BaseLayoutDetector, BaseVLRecognizer, BaseOCRRecognizer
 
 class ModelFactory: