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() ) ``` -result1 *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() ) ``` -result2
@@ -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)