From f784b4a63643ca28e9bc98f38c6d8263480a41dd Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Thu, 21 Apr 2022 17:32:00 +0800
Subject: [PATCH] Update

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 __init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 9637585..ca12ba5 100644
--- 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)