From ad84fcfce80af9c9046b504a148860904c7828f0 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: Tue, 7 May 2024 07:44:24 +0000 Subject: [PATCH] Added Language option in mixed mode --- src/web.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/src/web.py b/src/web.py index d4a84f9..9244fdc 100644 --- a/src/web.py +++ b/src/web.py @@ -111,6 +111,16 @@ with st.sidebar: on_change=change_side_bar ) + if inf_mode == "Text formula mixed": + lang = st.selectbox( + "Language", + ("English", "Chinese") + ) + if lang == "English": + lang = "en" + elif lang == "Chinese": + lang = "zh" + num_beams = st.number_input( 'Number of beams', min_value=1, @@ -124,7 +134,8 @@ with st.sidebar: ("cpu", "cuda", "mps"), on_change=change_side_bar ) - + + # ============================ end sidebar =============================== # @@ -214,7 +225,7 @@ elif uploaded_file or paste_result.image_data is not None: )[0] katex_res = to_katex(TexTeller_result) else: - katex_res = mix_inference(png_file_path, "en", infer_config, latex_det_model, lang_ocr_models, latex_rec_models, accelerator, num_beams) + katex_res = mix_inference(png_file_path, lang, infer_config, latex_det_model, lang_ocr_models, latex_rec_models, accelerator, num_beams) st.success('Completed!', icon="✅") st.markdown(suc_gif_html, unsafe_allow_html=True)