From a3250b6fcea1e9e03bcfda0eefb5f38add4d78ad Mon Sep 17 00:00:00 2001
From: felixhjh <852142024@qq.com>
Date: Mon, 29 Nov 2021 05:49:14 +0000
Subject: [PATCH] modify web_service to load gpu.yml and cpu.yml

---
 .../env_check/simple_web_service/web_service.py   | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

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 5f999f94..33e802e8 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()
-- 
GitLab