from speechbrain.pretrained import EncoderClassifier
import numpy as np

model = EncoderClassifier.from_hparams(
    source="speechbrain/spkrec-ecapa-voxceleb",
    run_opts={"device":"cpu"}
)

def compute_embedding(audio_path):
    signal, fs = model.load_audio(audio_path)
    embedding = model.encode_batch(signal.unsqueeze(0)).squeeze(0).detach().cpu().numpy()
    return embedding.astype(np.float32)