소스 검색

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