add package module for paddle_serving_app

python/paddle_serving_app/__init__.py

@@ -15,3 +15,4 @@ from .reader.chinese_bert_reader import ChineseBertReader
 from .reader.image_reader import ImageReader
 from .reader.lac_reader import LACReader
 from .reader.senta_reader import SentaReader
+from .models.model_list import ServingModels

python/paddle_serving_app/models/model_list.py

@@ -57,7 +57,7 @@ class ServingModels(object):
                 "resnet_v2_151_imagenet",
                 "resnet_v2_18_imagenet",
                 "resnet_v2_34_imagenet",
-                " resnet_v2_50_imagenet",
+                "resnet_v2_50_imagenet",
                 "resnext101_32x16d_wsl",
                 "resnext101_32x32d_wsl",
                 "resnext101_32x48d_wsl",

python/paddle_serving_app/package.py

+#   Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""
+Usage:
+    Download a package for serving directly
+    Example:
+        python -m paddle_serving_app.models --get senta_bilstm
+        python -m paddle_serving_app.models --list_model
+"""
+
+import argparse
+from .models import ServingModels
+
+
+def parse_args():  # pylint: disable=doc-string-missing
+    parser = argparse.ArgumentParser("serve")
+    parser.add_argument(
+        "--get", type=str, default="", help="Download a specific model")
+    parser.add_argument(
+        "--list_model", type=bool, default="", help="List Models")
+    return parser.parse_args()
+
+
+if __name__ == "__main__":
+    args = parse_args()
+    if args.list_model:
+        model_handle = ServingModels()
+        model_names = model_handle.get_model_list()
+        for key in model_names:
+            print(key)
+    elif args.get != "":
+        model_handle = ServingModels()
+        model_names = model_handle.get_model_list()
+        if args.get not in model_names:
+            print(
+                "Your model name does not exist in current model list, stay tuned"
+            )
+            sys.exit(0)
+        model_handle.download(args.get)
+    else:
+        print("Wrong argument")
+        print("""
+              Usage:
+              Download a package for serving directly
+              Example:
+                   python -m paddle_serving_app.models --get senta_bilstm
+                   python -m paddle_serving_app.models --list_model
+              """)
+        pass

python/setup.py.app.in

@@ -48,7 +48,8 @@ REQUIRED_PACKAGES = [
 packages=['paddle_serving_app',
           'paddle_serving_app.reader',
 	  'paddle_serving_app.utils',
-      'paddle_serving_app.reader.pddet']
+	  'paddle_serving_app.models',
+	  'paddle_serving_app.reader.pddet']
 
 package_data={}
 package_dir={'paddle_serving_app':
@@ -57,8 +58,10 @@ package_dir={'paddle_serving_app':
              '${PADDLE_SERVING_BINARY_DIR}/python/paddle_serving_app/reader',
 	     'paddle_serving_app.utils':
 	     '${PADDLE_SERVING_BINARY_DIR}/python/paddle_serving_app/utils',
-          'paddle_serving_app.reader.pddet':
-           '${PADDLE_SERVING_BINARY_DIR}/python/paddle_serving_app/reader/pddet',}
+	     'paddle_serving_app.models':
+	     '${PADDLE_SERVING_BINARY_DIR}/python/paddle_serving_app/models',
+	     'paddle_serving_app.reader.pddet':
+	     '${PADDLE_SERVING_BINARY_DIR}/python/paddle_serving_app/reader/pddet',}
 
 setup(
     name='paddle-serving-app',