Procházet zdrojové kódy

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

myhloli před 3 týdny
rodič
revize
701f6018f2

+ 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