|
|
--- |
|
|
license: apache-2.0 |
|
|
--- |
|
|
This is an [ECAPA model](https://huggingface.co/speechbrain/spkrec-ecapa-voxceleb) traced into a jit for simple use. |
|
|
|
|
|
Usage: |
|
|
|
|
|
```python |
|
|
from huggingface_hub import hf_hub_download |
|
|
import torch |
|
|
import soundfile as sf |
|
|
|
|
|
# Download the model from repo |
|
|
model_path = hf_hub_download( |
|
|
repo_id="balacoon/ecapa", |
|
|
filename="ecapa.jit", |
|
|
repo_type="model", |
|
|
local_dir="./", |
|
|
) |
|
|
|
|
|
# load model |
|
|
utmos_model = torch.jit.load(model_path).to(torch.device("cuda")) |
|
|
# load audio |
|
|
wav, sr = sf.read( |
|
|
"rms_arctic_a0001.wav", |
|
|
dtype="int16" |
|
|
) |
|
|
assert sr == 16000 |
|
|
# run inference |
|
|
x = torch.tensor(wav).unsqueeze(0).cuda() |
|
|
x_len = torch.tensor([x.shape[1]], device=x.device) |
|
|
emb = utmos_model(x, x_len) |
|
|
``` |
