Add yaml

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 0b2d9aa05a
1 changed files with 69 additions and 0 deletions
--- a/audio_embedding_clmr.yaml
+++ b/audio_embedding_clmr.yaml
@ -0,0 +1,69 @@
+name: 'audio_embedding_clmr'
+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/clmr-magnatagatune'
+        init_args:
+        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'