Spaces:
Build error
Build error
| 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 | |