From fd74966b272172aa862c3b34f1e2993b199f4e6b Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Tue, 21 Dec 2021 18:51:06 +0800
Subject: [PATCH] Add yaml

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 audio_embedding_vggish.yaml | 70 +++++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100644 audio_embedding_vggish.yaml

diff --git a/audio_embedding_vggish.yaml b/audio_embedding_vggish.yaml
new file mode 100644
index 0000000..52bba45
--- /dev/null
+++ b/audio_embedding_vggish.yaml
@@ -0,0 +1,70 @@
+name: 'audio_embedding_vggish'
+operators:
+    -
+        name: '_start_op'
+        function: '_start_op'
+        init_args:
+        inputs:
+            -
+                df: '_start_df'
+                name: 'filepath'
+                col: 0
+        outputs:
+            -
+                df: 'audio'
+        iter_info:
+            type: map
+    -
+        name: 'embedding_model'
+        function: 'towhee/tf_vggish_audioset'
+        init_args:
+            sample_rate: 16000
+        inputs:
+            -
+                df: 'audio'
+                name: 'filepath'
+                col: 0
+        outputs:
+            -
+                df: 'emb'
+        iter_info:
+            type: map
+    -
+        name: '_end_op'
+        function: '_end_op'
+        init_args:
+        inputs:
+            -
+                df: 'emb'
+                name: 'embs'
+                col: 0
+        outputs:
+            -
+                df: '_end_df'
+        iter_info:
+            type: map
+dataframes:
+    -
+        name: '_start_df'
+        columns:
+            -
+                name: 'filepath'
+                vtype: 'str'
+    -
+        name: 'audio'
+        columns:
+            -
+                name: 'filepath'
+                vtype: 'str'
+    -
+        name: 'emb'
+        columns:
+            -
+                name: 'embs'
+                vtype: 'numpy.ndarray'
+    -
+        name: '_end_df'
+        columns:
+            -
+                name: 'embs'
+                vtype: 'numpy.ndarray'