Update

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
3 years ago · f784b4a636
1 changed files with 2 additions and 2 deletions
--- a/init.py
+++ b/init.py
@ -15,5 +15,5 @@
 from .torch_vggish import Vggish


-def torch_vggish(weights_path: str = None, framework: str = 'pytorch'):
-    return Vggish(weights_path, framework)
+def torch_vggish(**kwargs):
+    return Vggish(kwargs)