Added Language option in mixed mode
This commit is contained in:
13
src/web.py
13
src/web.py
@@ -111,6 +111,16 @@ with st.sidebar:
|
|||||||
on_change=change_side_bar
|
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(
|
num_beams = st.number_input(
|
||||||
'Number of beams',
|
'Number of beams',
|
||||||
min_value=1,
|
min_value=1,
|
||||||
@@ -125,6 +135,7 @@ with st.sidebar:
|
|||||||
on_change=change_side_bar
|
on_change=change_side_bar
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
# ============================ end sidebar =============================== #
|
# ============================ end sidebar =============================== #
|
||||||
|
|
||||||
|
|
||||||
@@ -214,7 +225,7 @@ elif uploaded_file or paste_result.image_data is not None:
|
|||||||
)[0]
|
)[0]
|
||||||
katex_res = to_katex(TexTeller_result)
|
katex_res = to_katex(TexTeller_result)
|
||||||
else:
|
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.success('Completed!', icon="✅")
|
||||||
st.markdown(suc_gif_html, unsafe_allow_html=True)
|
st.markdown(suc_gif_html, unsafe_allow_html=True)
|
||||||
|
|||||||
Reference in New Issue
Block a user