Unverified Commit 1012dcb9 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #2857 from myhloli/dev

fix: add content existence checks for inline and interline equations in pipeline_middle_json_mkcontent.py
parents 97000c0b 46bfb774
......@@ -157,9 +157,11 @@ def merge_para_with_text(para_block):
if span_type == ContentType.TEXT:
content = escape_special_markdown_char(span['content'])
elif span_type == ContentType.INLINE_EQUATION:
content = f"{inline_left_delimiter}{span['content']}{inline_right_delimiter}"
if span.get('content', ''):
content = f"{inline_left_delimiter}{span['content']}{inline_right_delimiter}"
elif span_type == ContentType.INTERLINE_EQUATION:
content = f"\n{display_left_delimiter}\n{span['content']}\n{display_right_delimiter}\n"
if span.get('content', ''):
content = f"\n{display_left_delimiter}\n{span['content']}\n{display_right_delimiter}\n"
content = content.strip()
......
......@@ -225,7 +225,14 @@ def update_interface(backend_choice):
help="Enable gradio API for serving the application.",
default=True,
)
def main(example_enable, sglang_engine_enable, mem_fraction_static, torch_compile_enable, api_enable):
@click.option(
'--max-convert-pages',
'max_convert_pages',
type=int,
help="Set the maximum number of pages to convert from PDF to Markdown.",
default=1000,
)
def main(example_enable, sglang_engine_enable, mem_fraction_static, torch_compile_enable, api_enable, max_convert_pages):
if sglang_engine_enable:
try:
print("Start init SgLang engine...")
......@@ -257,7 +264,7 @@ def main(example_enable, sglang_engine_enable, mem_fraction_static, torch_compil
with gr.Row():
input_file = gr.File(label='Please upload a PDF or image', file_types=suffixes)
with gr.Row():
max_pages = gr.Slider(1, 20, 10, step=1, label='Max convert pages')
max_pages = gr.Slider(1, max_convert_pages, int(max_convert_pages/2), step=1, label='Max convert pages')
with gr.Row():
if sglang_engine_enable:
drop_list = ["pipeline", "vlm-sglang-engine"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment