提交 16ec3905 编写于 作者: D dongdaxiang

refine list_model

上级 fae303e5
...@@ -28,13 +28,13 @@ def parse_args(): # pylint: disable=doc-string-missing ...@@ -28,13 +28,13 @@ def parse_args(): # pylint: disable=doc-string-missing
parser.add_argument( parser.add_argument(
"--get_model", type=str, default="", help="Download a specific model") "--get_model", type=str, default="", help="Download a specific model")
parser.add_argument( parser.add_argument(
"--list_model", type=bool, default=False, help="List Models") '--list_model', nargs='*', default=None, help="List Models")
return parser.parse_args() return parser.parse_args()
if __name__ == "__main__": if __name__ == "__main__":
args = parse_args() args = parse_args()
if args.list_model: if args.list_model != None:
model_handle = ServingModels() model_handle = ServingModels()
model_names = model_handle.get_model_list() model_names = model_handle.get_model_list()
for key in model_names: for key in model_names:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册