From 19d97286e072b9016c70d11e1c3e965281911aa0 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Thu, 21 Apr 2022 17:27:44 +0800 Subject: [PATCH] Update Signed-off-by: Jael Gu --- __init__.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/__init__.py b/__init__.py index 37f5bd7..9637585 100644 --- a/__init__.py +++ b/__init__.py @@ -11,3 +11,9 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + +from .torch_vggish import Vggish + + +def torch_vggish(weights_path: str = None, framework: str = 'pytorch'): + return Vggish(weights_path, framework)