소스 검색

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
                     )