From 5d95d2e65c4950c8887b113f172f8a85cf112a55 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=89=E6=B4=8B=E4=B8=89=E6=B4=8B?= <1258009915@qq.com>
Date: Wed, 8 May 2024 14:18:44 +0000
Subject: [PATCH] bugfix

---
 src/models/ocr_model/utils/to_katex.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/models/ocr_model/utils/to_katex.py b/src/models/ocr_model/utils/to_katex.py
index eb63a7f..e7295bf 100644
--- a/src/models/ocr_model/utils/to_katex.py
+++ b/src/models/ocr_model/utils/to_katex.py
@@ -44,7 +44,7 @@ def change(input_str, old_inst, new_inst, old_surr_l, old_surr_r, new_surr_l, ne
                 i = j + 1
                 continue
             else:
-                assert count > 1
+                assert count >= 1
                 assert j == n
                 print("Warning: unbalanced surrogate pair in input string")
                 result += new_inst + new_surr_l
@@ -74,12 +74,13 @@ def to_katex(formula: str) -> str:
         r'\normalsize',
         r'\small',
         r'\footnotesize',
-        r'\scriptsize',
         r'\tiny'
     ]
     for (old_ins, new_ins) in zip(origin_instructions, origin_instructions):
         res = change(res, old_ins, new_ins, r'$', r'$', '{', '}')
+    res = change(res, r'\boldmath ', r'\bm', r'$', r'$', r'{', r'}')
     res = change(res, r'\boldmath', r'\bm', r'$', r'$', r'{', r'}')
+    res = change(res, r'\scriptsize', r'\scriptsize', r'$', r'$', r'{', r'}')
     
     origin_instructions = [
         r'\left',