Update examples/pipeline/bert

badb65a4 · TeslaZhao · 32715154 · badb65a4
隐藏空白更改
内联并排

Showing with 4 addition and 2 deletion

python/examples/pipeline/bert/web_service.py python/examples/pipeline/bert/web_service.py +4 -2

未找到文件。
--- a/python/examples/pipeline/bert/web_service.py
+++ b/python/examples/pipeline/bert/web_service.py
@@ -44,8 +44,10 @@ class BertOp(Op):
        return feed_dict, False, None, ""
    def postprocess(self, input_dicts, fetch_dict, data_id, log_id):
-        fetch_dict["pooled_output"] = str(fetch_dict["pooled_output"])
+        new_dict = {}
-        return fetch_dict, None, ""
+        new_dict["pooled_output"] = str(fetch_dict["pooled_output"])
+        new_dict["sequence_output"] = str(fetch_dict["sequence_output"])
+        return new_dict, None, ""
 class BertService(WebService):