소스 검색

fix: add logging for improved traceability in prediction logic of predict_formula.py

myhloli 3 주 전
부모
커밋
701f6018f2
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. 2 0
      mineru/model/mfr/pp_formulanet_plus_m/predict_formula.py

+ 2 - 0
mineru/model/mfr/pp_formulanet_plus_m/predict_formula.py

@@ -2,6 +2,8 @@ import os
 import torch
 import yaml
 from pathlib import Path
+
+from loguru import logger
 from tqdm import tqdm
 from mineru.model.utils.tools.infer import pytorchocr_utility
 from mineru.model.utils.pytorchocr.base_ocr_v20 import BaseOCRV20