Skip to content

GitLab

Commit 2cdfddd2 authored by zhougaofeng's avatar zhougaofeng
Browse files

Update ocr_server.py

parent e3c4beaf
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
magic_pdf/dict2md/ocr_server.py
View file @ 2cdfddd2
...@@ -26,6 +26,8 @@ REVISION = 'v1.0.4' ...@@ -26,6 +26,8 @@ REVISION = 'v1.0.4'
BOX_TAG_PATTERN = r"<box>([\s\S]*?)</box>" BOX_TAG_PATTERN = r"<box>([\s\S]*?)</box>"
PUNCTUATION = "！？。＂＃＄％＆＇（）＊＋，－／：；＜＝＞＠［＼］＾＿｀｛｜｝～｟｠｢｣､、〃》「」『』【】〔〕〖〗〘〙〚〛〜〝〞〟〰〾〿–—‘’‛“”„‟…‧﹏." PUNCTUATION = "！？。＂＃＄％＆＇（）＊＋，－／：；＜＝＞＠［＼］＾＿｀｛｜｝～｟｠｢｣､、〃》「」『』【】〔〕〖〗〘〙〚〛〜〝〞〟〰〾〿–—‘’‛“”„‟…‧﹏."
logger.add("../../parse.log", rotation="10 MB", level="INFO",
format="{time} {level} {message}", encoding='utf-8', enqueue=True)
def _get_args(): def _get_args():
parser = ArgumentParser() parser = ArgumentParser()
...@@ -253,10 +255,8 @@ if __name__ == "__main__": ...@@ -253,10 +255,8 @@ if __name__ == "__main__":
args = _get_args() args = _get_args()
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read(args.config_path) config.read(args.config_path)
ocr_server = config.get('server', 'ocr_server') host = config.get('server', 'ocr_host')
if 'http' in ocr_server: port = int(config.get('server', 'ocr_port'))
ocr_server = ocr_server.split('://')[1]
host,port = ocr_server.split(':')[0],int(ocr_server.split(':')[1])
uvicorn.run(app, host=host, port=port) uvicorn.run(app, host=host, port=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