Explorar o código

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

myhloli hai 3 semanas
pai
achega
701f6018f2
Modificáronse 1 ficheiros con 2 adicións e 0 borrados
  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