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)