提交 de926182 编写于 作者: T TeslaZhao

fix bad code style by yapf

上级 1c41772b
...@@ -24,58 +24,50 @@ from flask import Flask, request ...@@ -24,58 +24,50 @@ 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, type=str,
help="Port the server") default="workdir",
parser.add_argument("--name", help="Working dir of current service")
type=str, parser.add_argument(
default="None", "--device", type=str, default="cpu", help="Type of device")
help="Web service name") parser.add_argument(
parser.add_argument("--workdir", "--mem_optim_off",
type=str, default=False,
default="workdir", action="store_true",
help="Working dir of current service") help="Memory optimize")
parser.add_argument("--device", parser.add_argument(
type=str, "--ir_optim", default=False, action="store_true", help="Graph optimize")
default="cpu", parser.add_argument(
help="Type of device") "--use_mkl", default=False, action="store_true", help="Use MKL")
parser.add_argument("--mem_optim_off", parser.add_argument(
default=False, "--max_body_size",
action="store_true", type=int,
help="Memory optimize") default=512 * 1024 * 1024,
parser.add_argument("--ir_optim", help="Limit sizes of messages")
default=False, parser.add_argument(
action="store_true", "--use_multilang",
help="Graph optimize") default=False,
parser.add_argument("--use_mkl", action="store_true",
default=False, help="Use Multi-language-service")
action="store_true", parser.add_argument(
help="Use MKL") "--product_name",
parser.add_argument("--max_body_size", type=str,
type=int, default=None,
default=512 * 1024 * 1024, help="product_name for authentication")
help="Limit sizes of messages") parser.add_argument(
parser.add_argument("--use_multilang", "--container_id",
default=False, type=str,
action="store_true", default=None,
help="Use Multi-language-service") help="container_id for authentication")
parser.add_argument("--product_name",
type=str,
default=None,
help="product_name for authentication")
parser.add_argument("--container_id",
type=str,
default=None,
help="container_id for authentication")
return parser.parse_args() return parser.parse_args()
...@@ -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.
先完成此消息的编辑!
想要评论请 注册