From 1464ec2993f835bd034ad0238aeb3cee1bff2386 Mon Sep 17 00:00:00 2001 From: dongdaxiang Date: Wed, 6 May 2020 16:58:21 +0800 Subject: [PATCH] add tutorial function --- python/paddle_serving_app/models/model_list.py | 4 ++++ python/paddle_serving_app/package.py | 12 ++++++++++++ 2 files changed, 16 insertions(+) diff --git a/python/paddle_serving_app/models/model_list.py b/python/paddle_serving_app/models/model_list.py index c2638f30..a2019997 100644 --- a/python/paddle_serving_app/models/model_list.py +++ b/python/paddle_serving_app/models/model_list.py @@ -59,6 +59,10 @@ class ServingModels(object): url = self.url_dict[model_name] r = os.system('wget ' + url + ' --no-check-certificate') + def get_tutorial(self, model_name): + if model_name in self.tutorial_url: + return "Tutorial of {} to be added".format(model_name) + if __name__ == "__main__": models = ServingModels() diff --git a/python/paddle_serving_app/package.py b/python/paddle_serving_app/package.py index 5218f63f..e2791493 100644 --- a/python/paddle_serving_app/package.py +++ b/python/paddle_serving_app/package.py @@ -30,6 +30,8 @@ def parse_args(): # pylint: disable=doc-string-missing "--get_model", type=str, default="", help="Download a specific model") parser.add_argument( '--list_model', nargs='*', default=None, help="List Models") + parser.add_argument( + '--tutorial', type=str, default="", help="Get running command") return parser.parse_args() @@ -54,6 +56,16 @@ if __name__ == "__main__": ) sys.exit(0) model_handle.download(args.get_model) + elif args.tutorial != "": + model_handle = ServingModels() + model_dict = model_handle.url_dict + if args.get_model not in model_dict: + print( + "Your model name does not exist in current model list, stay tuned" + ) + sys.exit(0) + tutorial_str = model_handle.get_tutorial() + print(tutorial_str) else: print("Wrong argument") print(""" -- GitLab