Update README.md
Browse files
README.md
CHANGED
|
@@ -33,7 +33,6 @@ Now you can download the model by executing the following code:
|
|
| 33 |
from huggingface_hub import hf_hub_download
|
| 34 |
|
| 35 |
model_file = hf_hub_download(repo_id=Jenthe/ECAPA2, filename='model.pt')
|
| 36 |
-
model = torch.jit.load(model_file, map_location='cpu')
|
| 37 |
```
|
| 38 |
|
| 39 |
Subsequent calls will load the previously downloaded model automatically.
|
|
@@ -46,6 +45,7 @@ import torch
|
|
| 46 |
import torchaudio
|
| 47 |
|
| 48 |
audio = torchaudio.load('sample.wav')
|
|
|
|
| 49 |
embedding = model.extract_embedding(audio)
|
| 50 |
```
|
| 51 |
|
|
|
|
| 33 |
from huggingface_hub import hf_hub_download
|
| 34 |
|
| 35 |
model_file = hf_hub_download(repo_id=Jenthe/ECAPA2, filename='model.pt')
|
|
|
|
| 36 |
```
|
| 37 |
|
| 38 |
Subsequent calls will load the previously downloaded model automatically.
|
|
|
|
| 45 |
import torchaudio
|
| 46 |
|
| 47 |
audio = torchaudio.load('sample.wav')
|
| 48 |
+
model = torch.jit.load(model_file, map_location='cpu')
|
| 49 |
embedding = model.extract_embedding(audio)
|
| 50 |
```
|
| 51 |
|