change web service

python/paddle_serving_server/web_service.py

@@ -55,22 +55,18 @@ class WebService(object):
             "{}/serving_server_conf.prototxt".format(self.model_config))
         self.client_service.connect(["0.0.0.0:{}".format(self.port + 1)])
 
-        @app_instance.route(service_name, methods=['POST'])
     def get_prediction():
         if not request.json:
             abort(400)
         if "fetch" not in request.json:
             abort(400)
         try:
-                feed, fetch = self.preprocess(request.json,
-                                              request.json["fetch"])
+            feed, fetch = self.preprocess(request.json, request.json["fetch"])
             if isinstance(feed, list):
                 fetch_map_batch = client_service.predict(
                     feed_batch=feed, fetch=fetch)
                 fetch_map_batch = self.postprocess(
-                        feed=request.json,
-                        fetch=fetch,
-                        fetch_map=fetch_map_batch)
+                    feed=request.json, fetch=fetch, fetch_map=fetch_map_batch)
                 for key in fetch_map_batch:
                     fetch_map_batch[key] = fetch_map_batch[key].tolist()
                 result = {"result": fetch_map_batch}
@@ -86,11 +82,6 @@ class WebService(object):
             result = {"result": "Request Value Error"}
         return result
 
-        app_instance.run(host="0.0.0.0",
-                         port=self.port,
-                         threaded=False,
-                         processes=1)
-
     def run_server(self):
         import socket
         localIP = socket.gethostbyname(socket.gethostname())

python/paddle_serving_server_gpu/web_service.py

@@ -147,7 +147,6 @@ class WebService(object):
 
         self.idx = 0
 
-        @app_instance.route(service_name, methods=['POST'])
     def get_prediction():
         if not request.json:
             abort(400)
@@ -164,14 +163,6 @@ class WebService(object):
             result = {"result": "Request Value Error"}
         return result
 
-        app_instance.run(host="0.0.0.0",
-                         port=self.port,
-                         threaded=False,
-                         processes=1)
-
-        for p in producer_list:
-            p.join()
-
     def run_server(self):
         import socket
         localIP = socket.gethostbyname(socket.gethostname())