diff --git a/python/paddle_serving_server/env_check/simple_web_service/web_service.py b/python/paddle_serving_server/env_check/simple_web_service/web_service.py index 5f999f94f9da10809c0128a45c115d90f05f0f41..33e802e816231618d8c474f56cfcf6ed350adf83 100644 --- a/python/paddle_serving_server/env_check/simple_web_service/web_service.py +++ b/python/paddle_serving_server/env_check/simple_web_service/web_service.py @@ -16,9 +16,20 @@ from paddle_serving_server.web_service import WebService, Op import logging import numpy as np import sys +import argparse _LOGGER = logging.getLogger() +def serve_args(): + parser = argparse.ArgumentParser("serve") + parser.add_argument( + "config", + type=str, + default="config.yml", + nargs="?", + help="cpu or gpu config.yml") + return parser.parse_args() + class UciOp(Op): def init_op(self): @@ -53,7 +64,7 @@ class UciService(WebService): uci_op = UciOp(name="uci", input_ops=[read_op]) return uci_op - +args = serve_args() uci_service = UciService(name="uci") -uci_service.prepare_pipeline_config("config.yml") +uci_service.prepare_pipeline_config(args.config) uci_service.run_service()