Browse Source

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

myhloli 3 tuần trước cách đây
mục cha
commit
701f6018f2
1 tập tin đã thay đổi với 2 bổ sung0 xóa
  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