未验证 提交 302d7ebc 编写于 作者: J Jiawei Wang 提交者: GitHub

Merge pull request #1108 from zhangjun/py-fix

debug_service fix
...@@ -107,7 +107,7 @@ ocr_service.prepare_server(workdir="workdir", port=9292) ...@@ -107,7 +107,7 @@ ocr_service.prepare_server(workdir="workdir", port=9292)
ocr_service.init_det_debugger(det_model_config="ocr_det_model") ocr_service.init_det_debugger(det_model_config="ocr_det_model")
if sys.argv[1] == 'gpu': if sys.argv[1] == 'gpu':
ocr_service.set_gpus("2") ocr_service.set_gpus("2")
ocr_service.run_debugger_service() ocr_service.run_debugger_service(gpu = True)
elif sys.argv[1] == 'cpu': elif sys.argv[1] == 'cpu':
ocr_service.run_debugger_service() ocr_service.run_debugger_service()
ocr_service.run_web_service() ocr_service.run_web_service()
...@@ -266,8 +266,12 @@ class WebService(object): ...@@ -266,8 +266,12 @@ class WebService(object):
def _launch_local_predictor(self, gpu): def _launch_local_predictor(self, gpu):
from paddle_serving_app.local_predict import LocalPredictor from paddle_serving_app.local_predict import LocalPredictor
self.client = LocalPredictor() self.client = LocalPredictor()
self.client.load_model_config( if gpu:
"{}".format(self.model_config), use_gpu=True, gpu_id=self.gpus[0]) self.client.load_model_config(
"{}".format(self.model_config), use_gpu=True, gpu_id=self.gpus[0])
else:
self.client.load_model_config(
"{}".format(self.model_config), use_gpu=False)
def run_web_service(self): def run_web_service(self):
print("This API will be deprecated later. Please do not use it") print("This API will be deprecated later. Please do not use it")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册