浏览代码

fix: replace hardcoded model name with constant in batch_analyze.py

myhloli 3 月之前
父节点
当前提交
0298de844f
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      mineru/backend/pipeline/batch_analyze.py

+ 1 - 1
mineru/backend/pipeline/batch_analyze.py

@@ -340,7 +340,7 @@ class BatchAnalyze:
                     # Get OCR results for this language's images
 
                     ocr_model = atom_model_manager.get_atom_model(
-                        atom_model_name='ocr',
+                        atom_model_name=AtomicModel.OCR,
                         det_db_box_thresh=0.3,
                         lang=lang
                     )