diff --git a/README.md b/README.md
index 8db2f05..e409733 100644
--- a/README.md
+++ b/README.md
@@ -25,11 +25,11 @@ import towhee
towhee.glob('./towhee.jpg') \
.image_decode.cv2() \
- .image_embedding.data2vec_vision(model_name='facebook/data2vec-vision-base-ft1k') \
+ .image_embedding.data2vec(model_name='facebook/data2vec-vision-base-ft1k') \
.show()
```
-
+
*Write a same pipeline with explicit inputs/outputs name specifications:*
@@ -38,11 +38,11 @@ import towhee
towhee.glob['path']('./towhee.jpg') \
.image_decode.cv2['path', 'img']() \
- .image_embedding.data2vec_vision['img', 'vec'](model_name='facebook/data2vec-vision-base-ft1k') \
+ .image_embedding.data2vec['img', 'vec'](model_name='facebook/data2vec-vision-base-ft1k') \
.select['img', 'vec']() \
.show()
```
-
+
@@ -53,7 +53,7 @@ towhee.glob['path']('./towhee.jpg') \
Create the operator via the following factory method
-***data2vec_vision(model_name='facebook/data2vec-vision-base')***
+***data2vec(model_name='facebook/data2vec-vision-base')***
**Parameters:**
diff --git a/__init__.py b/__init__.py
index 5fd45d1..0559bc2 100644
--- a/__init__.py
+++ b/__init__.py
@@ -15,5 +15,5 @@
from .data2vec_vision import Data2VecVision
-def data2vec_vision(model_name='facebook/data2vec-vision-base'):
+def data2vec(model_name='facebook/data2vec-vision-base'):
return Data2VecVision(model_name)