centersl/MinerU
1
0
Fork 0
mirror of https://github.com/opendatalab/MinerU.git synced 2026-03-27 11:08:32 +07:00
Code Issues Packages Projects Releases Wiki Activity

fix: update progress bar description in formula prediction

Browse Source
This commit is contained in:
myhloli
2025-10-22 15:51:56 +08:00
parent f57a6d8d9e
commit ab480a7a86
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mineru/model/mfr/pp_formulanet_plus_m/predict_formula.py
View File

@@ -73,7 +73,7 @@ class FormulaRecognizer(BaseOCRV20):
inp = inp.to(self.device)
rec_formula = []
with torch.no_grad():
with tqdm(total=len(inp), desc="Formula Predict") as pbar:
with tqdm(total=len(inp), desc="MFR Predict") as pbar:
for index in range(0, len(inp), batch_size):
batch_data = inp[index: index + batch_size]
batch_preds = [self.net(batch_data)]