Parcourir la source

feat: 更新VL识别器创建逻辑,支持Paddle适配器

zhch158_admin il y a 2 semaines
Parent
commit
b13ce17c58
1 fichiers modifiés avec 4 ajouts et 1 suppressions
  1. 4 1
      zhch/universal_doc_parser/core/model_factory.py

+ 4 - 1
zhch/universal_doc_parser/core/model_factory.py

@@ -50,7 +50,10 @@ class ModelFactory:
         """创建VL识别器"""
         module_name = config.get('module', 'mineru')
         
-        if module_name == 'mineru':
+        if module_name == 'paddle':
+            from models.adapters import PaddleVLRecognizer
+            recognizer = PaddleVLRecognizer(config)
+        elif module_name == 'mineru':
             from models.adapters import MinerUVLRecognizer
             recognizer = MinerUVLRecognizer(config)
         else: