diff --git a/python/examples/bert/README.md b/python/examples/bert/README.md index 612ba8beb979f5a1ee9d1a8052f73169191f4fd7..acbd3d05cd4197e0e74f9ab04b141488145094ec 100644 --- a/python/examples/bert/README.md +++ b/python/examples/bert/README.md @@ -42,7 +42,7 @@ pip install paddle_serving_app ``` 执行 ``` -cat data-c.txt | python bert_client.py +head data-c.txt | python bert_client.py --model bert_seq20_client/serving_client_conf.prototxt ``` 启动client读取data-c.txt中的数据进行预测,预测结果为文本的向量表示(由于数据较多,脚本中没有将输出进行打印),server端的地址在脚本中修改。 diff --git a/python/examples/bert/bert_client.py b/python/examples/bert/bert_client.py index cceb07c7334aa0cc9b9c347ae53351b7a6fa3513..51364c6745731017b31923d246990497115dc780 100644 --- a/python/examples/bert/bert_client.py +++ b/python/examples/bert/bert_client.py @@ -29,13 +29,13 @@ from paddle_serving_app import ChineseBertReader args = benchmark_args() -reader = ChineseBertReader(max_seq_len=20) +reader = ChineseBertReader({"max_seq_len": 20}) fetch = ["pooled_output"] endpoint_list = ["127.0.0.1:9292"] client = Client() client.load_client_config(args.model) client.connect(endpoint_list) -for line in fin: +for line in sys.stdin: feed_dict = reader.process(line) result = client.predict(feed=feed_dict, fetch=fetch)