瀏覽代碼

fix: update progress bar description in formula prediction

myhloli 3 周之前
父節點
當前提交
ab480a7a86
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      mineru/model/mfr/pp_formulanet_plus_m/predict_formula.py

+ 1 - 1
mineru/model/mfr/pp_formulanet_plus_m/predict_formula.py

@@ -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)]