Ver código fonte

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

myhloli 3 meses atrás
pai
commit
0298de844f
1 arquivos alterados com 1 adições e 1 exclusões
  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
                     )