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)