提交 0e84c13d 编写于 作者: T TeslaZhao

fix bad code style by yapf

上级 e6263b1f
...@@ -24,55 +24,47 @@ from flask import Flask, request ...@@ -24,55 +24,47 @@ from flask import Flask, request
def parse_args(): # pylint: disable=doc-string-missing def parse_args(): # pylint: disable=doc-string-missing
parser = argparse.ArgumentParser("serve") parser = argparse.ArgumentParser("serve")
parser.add_argument("--thread", parser.add_argument(
type=int, "--thread", type=int, default=10, help="Concurrency of server")
default=10, parser.add_argument(
help="Concurrency of server") "--model", type=str, default="", help="Model for serving")
parser.add_argument("--model", parser.add_argument(
type=str, "--port", type=int, default=9292, help="Port the server")
default="", parser.add_argument(
help="Model for serving") "--name", type=str, default="None", help="Web service name")
parser.add_argument("--port", parser.add_argument(
type=int, "--workdir",
default=9292,
help="Port the server")
parser.add_argument("--name",
type=str,
default="None",
help="Web service name")
parser.add_argument("--workdir",
type=str, type=str,
default="workdir", default="workdir",
help="Working dir of current service") help="Working dir of current service")
parser.add_argument("--device", parser.add_argument(
type=str, "--device", type=str, default="cpu", help="Type of device")
default="cpu", parser.add_argument(
help="Type of device") "--mem_optim_off",
parser.add_argument("--mem_optim_off",
default=False, default=False,
action="store_true", action="store_true",
help="Memory optimize") help="Memory optimize")
parser.add_argument("--ir_optim", parser.add_argument(
default=False, "--ir_optim", default=False, action="store_true", help="Graph optimize")
action="store_true", parser.add_argument(
help="Graph optimize") "--use_mkl", default=False, action="store_true", help="Use MKL")
parser.add_argument("--use_mkl", parser.add_argument(
default=False, "--max_body_size",
action="store_true",
help="Use MKL")
parser.add_argument("--max_body_size",
type=int, type=int,
default=512 * 1024 * 1024, default=512 * 1024 * 1024,
help="Limit sizes of messages") help="Limit sizes of messages")
parser.add_argument("--use_multilang", parser.add_argument(
"--use_multilang",
default=False, default=False,
action="store_true", action="store_true",
help="Use Multi-language-service") help="Use Multi-language-service")
parser.add_argument("--product_name", parser.add_argument(
"--product_name",
type=str, type=str,
default=None, default=None,
help="product_name for authentication") help="product_name for authentication")
parser.add_argument("--container_id", parser.add_argument(
"--container_id",
type=str, type=str,
default=None, default=None,
help="container_id for authentication") help="container_id for authentication")
...@@ -137,9 +129,8 @@ if __name__ == "__main__": ...@@ -137,9 +129,8 @@ if __name__ == "__main__":
else: else:
service = WebService(name=args.name) service = WebService(name=args.name)
service.load_model_config(args.model) service.load_model_config(args.model)
service.prepare_server(workdir=args.workdir, service.prepare_server(
port=args.port, workdir=args.workdir, port=args.port, device=args.device)
device=args.device)
service.run_rpc_service() service.run_rpc_service()
app_instance = Flask(__name__) app_instance = Flask(__name__)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册