Skip to content

GitLab

Commit 8a1b6709 authored by zhougaofeng's avatar zhougaofeng
Browse files

Update ocr_vllm_server.py

parent 8196f3c7
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
magic_pdf/dict2md/ocr_vllm_server.py
View file @ 8a1b6709
......@@ -285,6 +285,8 @@ if __name__ == "__main__":
args = get_args()
config = configparser.ConfigParser()
config.read(args.config_path)
host = config.get('server', 'ocr_host')
port = int(config.get('server', 'ocr_port'))
# host = config.get('server', 'ocr_host')
# port = int(config.get('server', 'ocr_port'))
host, port = config.get('server', 'ocr_server').split('://')[1].split(':')[0], int(
config.get('server', 'ocr_server').split('://')[1].split(':')[1])
uvicorn.run(app, host=args.host, port=args.port)
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