You can try using tensorflow model hub

import tensorflow as tf

import tensorflow_hub as hub

import numpy as np

import librosa

# Load the model.

vggmodel = hub.load('https://tfhub.dev/google/vggish/1')

def embedding_from_fn(fn):

x, sr = librosa.load(fn,sr=None)

x_16k = librosa.resample(x,sr,16000) #resample to 16KHz

embedding = np.array(vggmodel(x_16k))

return embedding

embedding = embedding_from_fn(df['fn'].values[0]).shape