From 556d4a26e01ee372ff890f75ff0841db7b18bcac Mon Sep 17 00:00:00 2001
From: wxywb <xy.wang@zilliz.com>
Date: Fri, 24 Jun 2022 18:11:53 +0800
Subject: [PATCH] change data2vec_audio to data2vec.

Signed-off-by: wxywb <xy.wang@zilliz.com>
---
 README.md   | 8 +++-----
 __init__.py | 2 +-
 2 files changed, 4 insertions(+), 6 deletions(-)

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()
 )
 
 ```
-<img src="https://towhee.io/towhee/data2vec-vision/raw/branch/main/result1.png" alt="result1" style="height:20px;"/>
 
 *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()
 )
 ```
-<img src="https://towhee.io/towhee/data2vec-vision/raw/branch/main/result2.png" alt="result2" style="height:60px;"/>
 
 
 <br />
@@ -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)