Mirrors/dots.ocr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updata inference demo

Browse Source
This commit is contained in:
zhd36
2026-02-16 02:48:00 +08:00
parent c3084bd437
commit b06a7810ac
7 changed files with 108 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
demo/demo_vllm.py
+9 -1
View File
@@ -11,7 +11,15 @@ parser = argparse.ArgumentParser()
parser.add_argument("--ip", type=str, default="localhost")
parser.add_argument("--port", type=str, default="8000")
parser.add_argument("--model_name", type=str, default="rednote-hilab/dots.ocr")
parser.add_argument("--prompt_mode", type=str, default="prompt_layout_all_en")
parser.add_argument("--prompt_mode", type=str, default="prompt_layout_all_en",help=(
"Choose a task prompt: "
"prompt_layout_all_en=full document layout+OCR to JSON/MD; "
"prompt_layout_only_en=layout detection only; "
"prompt_grounding_ocr=OCR within a given bbox; "
"prompt_web_parsing=parse webpage screenshot layout into JSON; "
"prompt_scene_spotting=detect+recognize scene text (OCR boxes+texts); "
"prompt_image_to_svg=generate SVG code to reconstruct the image.")
)
args = parser.parse_args()