diff --git a/README.md b/README.md
index 86effb6..5382860 100644
--- a/README.md
+++ b/README.md
@@ -28,12 +28,11 @@ import towhee
towhee.glob('test.wav')
.audio_decode.ffmpeg()
.runas_op(func=lambda x:[y[0] for y in x])
- .towhee.data2vec_audio()
+ .towhee.data2vec()
.show()
)
```
-
*Write a same pipeline with explicit inputs/outputs name specifications:*
@@ -44,11 +43,10 @@ import towhee
towhee.glob['path']('test.wav')
.audio_decode.ffmpeg['path', 'frames']()
.runas_op['frames', 'frames'](func=lambda x:[y[0] for y in x])
- .towhee.data2vec_audio['frames', 'vecs'](model_name="facebook/data2vec-audio-base-960h")
+ .towhee.data2vec['frames', 'vecs'](model_name="facebook/data2vec-audio-base-960h")
.show()
)
```
-
@@ -59,7 +57,7 @@ import towhee
Create the operator via the following factory method
-***data2vec_vision(model_name='facebook/data2vec-vision-base')***
+***data2vec(model_name='facebook/data2vec-audio-base')***
**Parameters:**
diff --git a/__init__.py b/__init__.py
index c56fe53..04cfa73 100644
--- a/__init__.py
+++ b/__init__.py
@@ -14,5 +14,5 @@
from .data2vec_audio import Data2VecAudio
-def data2vec_audio(model_name="facebook/data2vec-audio-base-960h"):
+def data2vec(model_name="facebook/data2vec-audio-base-960h"):
return Data2VecAudio(model_name)