YogeLiu/TexTeller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Modify the names of options in the web.py

Browse Source 
Formula only       -> Formula recognition
Text formula mixed -> Paragraph recognition

Improved display during mixed inference
This commit is contained in:
三洋三洋
2024-06-04 14:23:56 +00:00
parent dbbec511ef
commit 2920b753a8
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/web.py
View File

@@ -110,10 +110,10 @@ if "CHANGE_SIDEBAR_FLAG" not in st.session_state:
st.session_state["CHANGE_SIDEBAR_FLAG"] = False st.session_state["CHANGE_SIDEBAR_FLAG"] = False
if "INF_MODE" not in st.session_state: if "INF_MODE" not in st.session_state:
st.session_state["INF_MODE"] = "Formula only" st.session_state["INF_MODE"] = "Formula recognition"
# ============================ begin sidebar =============================== # ############################## <sidebar> ##############################
with st.sidebar: with st.sidebar:
num_beams = 1 num_beams = 1
@@ -123,7 +123,7 @@ with st.sidebar:
inf_mode = st.selectbox( inf_mode = st.selectbox(
"Inference mode", "Inference mode",
("Formula only", "Text formula mixed"), ("Formula recognition", "Paragraph recognition"),
on_change=change_side_bar on_change=change_side_bar
) )
@@ -142,16 +142,16 @@ with st.sidebar:
) )
# ============================ end sidebar =============================== # ############################## </sidebar> ##############################
# ============================ begin pages =============================== # ################################ <page> ################################
texteller = get_texteller() texteller = get_texteller()
tokenizer = get_tokenizer() tokenizer = get_tokenizer()
latex_rec_models = [texteller, tokenizer] latex_rec_models = [texteller, tokenizer]
if inf_mode == "Text formula mixed": if inf_mode == "Paragraph recognition":
infer_config, latex_det_model = get_det_models() infer_config, latex_det_model = get_det_models()
lang_ocr_models = get_ocr_models(accelerator) lang_ocr_models = get_ocr_models(accelerator)
@@ -221,7 +221,7 @@ elif uploaded_file or paste_result.image_data is not None:
st.write("") st.write("")
with st.spinner("Predicting..."): with st.spinner("Predicting..."):
if inf_mode == "Formula only": if inf_mode == "Formula recognition":
TexTeller_result = latex_recognition( TexTeller_result = latex_recognition(
texteller, texteller,
tokenizer, tokenizer,
@@ -237,13 +237,13 @@ elif uploaded_file or paste_result.image_data is not None:
st.markdown(suc_gif_html, unsafe_allow_html=True) st.markdown(suc_gif_html, unsafe_allow_html=True)
st.text_area(":blue[*** 𝑃r𝑒d𝑖c𝑡e𝑑 𝑓o𝑟m𝑢l𝑎 ***]", katex_res, height=150) st.text_area(":blue[*** 𝑃r𝑒d𝑖c𝑡e𝑑 𝑓o𝑟m𝑢l𝑎 ***]", katex_res, height=150)
if inf_mode == "Formula only": if inf_mode == "Formula recognition":
st.latex(katex_res) st.latex(katex_res)
elif inf_mode == "Text formula mixed": elif inf_mode == "Paragraph recognition":
mixed_res = re.split(r'(\n\$\$.*?\$\$\n)', katex_res) mixed_res = re.split(r'(\$\$.*?\$\$)', katex_res)
for text in mixed_res: for text in mixed_res:
if text.startswith('\n$$') and text.endswith('$$\n'): if text.startswith('$$') and text.endswith('$$'):
st.latex(text[3:-3]) st.latex(text[2:-2])
else: else:
st.markdown(text) st.markdown(text)
@@ -262,4 +262,4 @@ elif uploaded_file or paste_result.image_data is not None:
paste_result.image_data = None paste_result.image_data = None
# ============================ end pages =============================== # ################################ </page> ################################