From cc78bd12f4a7e3d76ce669cdbff28906d3c9ae43 Mon Sep 17 00:00:00 2001 From: ShenYuhan Date: Tue, 11 Aug 2020 11:42:08 +0800 Subject: [PATCH] print traceback for serving (#815) --- paddlehub/commands/tmpl/x_model.tmpl | 6 +++--- paddlehub/serving/app_single.py | 7 +++---- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/paddlehub/commands/tmpl/x_model.tmpl b/paddlehub/commands/tmpl/x_model.tmpl index 26fae122..4d63a6f8 100644 --- a/paddlehub/commands/tmpl/x_model.tmpl +++ b/paddlehub/commands/tmpl/x_model.tmpl @@ -20,9 +20,9 @@ def base64_to_cv2(b64str): def cv2_to_base64(image): - return base64.b64encode(image) - # data = cv2.imencode('.jpg', image)[1] - # return base64.b64encode(data.tostring()).decode('utf8') + # return base64.b64encode(image) + data = cv2.imencode('.jpg', image)[1] + return base64.b64encode(data.tostring()).decode('utf8') def read_images(paths): diff --git a/paddlehub/serving/app_single.py b/paddlehub/serving/app_single.py index 7bcbe12e..0d288de7 100644 --- a/paddlehub/serving/app_single.py +++ b/paddlehub/serving/app_single.py @@ -17,6 +17,7 @@ from paddlehub.serving.model_service.base_model_service import nlp_module_info from paddlehub.serving.model_service.base_model_service import v2_module_info from paddlehub.common import utils import functools +import traceback import time import os import base64 @@ -40,8 +41,7 @@ def predict_v2_advanced(module_info, input): try: output = serving_method(**predict_args) except Exception as err: - curr = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(time.time())) - print(curr, " - ", err) + traceback.print_exc() return gen_result("-1", "Please check data format!", "") return gen_result("0", "", output) @@ -66,8 +66,7 @@ def predict_nlp(module_info, input_text, req_id, extra=None): try: res = predict_method(**predict_args) except Exception as err: - curr = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(time.time())) - print(curr, " - ", err) + traceback.print_exc() return gen_result("-1", "Please check data format!", "") finally: user_dict = extra.get("user_dict", []) -- GitLab