From 30ce95b71527ba92d9558ae30a5f48926b337679 Mon Sep 17 00:00:00 2001 From: shiyu22 Date: Wed, 14 Jun 2023 14:24:09 +0800 Subject: [PATCH] Update README --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index e185cbf..2bb96b6 100644 --- a/README.md +++ b/README.md @@ -21,7 +21,7 @@ from towhee import pipe, ops, DataCollection p = pipe.input('collection_name', 'text') \ .map('text', 'vec', ops.sentence_embedding.transformers(model_name='all-MiniLM-L12-v2')) \ - .flat_map(('collection_name', 'vec'), 'rows', ops.ann_search.milvus_client(host='127.0.0.1', port='19530', **{'output_fields': ['text']})) \ + .map(('collection_name', 'vec'), 'rows', ops.ann_search.osschat_milvus(host='127.0.0.1', port='19530', **{'output_fields': ['text']})) \ .map('rows', ('id', 'score', 'text'), lambda x: (x[0], x[1], x[2])) \ .output('id', 'score', 'text') @@ -39,7 +39,7 @@ from towhee import pipe, ops, DataCollection p = pipe.input('collection_name', 'text') \ .map('text', 'vec', ops.sentence_embedding.transformers(model_name='all-MiniLM-L12-v2')) \ - .map(('collection_name', 'vec'), 'rows', ops.ann_search.milvus_client(host='127.0.0.1', port='19530', **{'output_fields': ['text']})) \ + .map(('collection_name', 'vec'), 'rows', ops.ann_search.osschat_milvus(host='127.0.0.1', port='19530', **{'output_fields': ['text']})) \ .output('rows') DataCollection(p('test_collection', 'cat')).show()