Spaces:
Build error
Build error
File size: 446 Bytes
b0c0f32 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
dependencies = ['torch', 'numpy', 'resampy', 'soundfile']
from torchvggish.vggish import VGGish
model_urls = {
'vggish': 'https://github.com/harritaylor/torchvggish/'
'releases/download/v0.1/vggish-10086976.pth',
'pca': 'https://github.com/harritaylor/torchvggish/'
'releases/download/v0.1/vggish_pca_params-970ea276.pth'
}
def vggish(**kwargs):
model = VGGish(urls=model_urls, **kwargs)
return model
|