Update README.md
Browse files
README.md
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
---
|
2 |
language:
|
3 |
- en
|
4 |
-
datasets:
|
5 |
tags:
|
6 |
- speech
|
7 |
---
|
@@ -30,13 +29,13 @@ The model is fine-tuned on the [VoxCeleb1 dataset](https://www.robots.ox.ac.uk/~
|
|
30 |
# Usage
|
31 |
## Speaker Verification
|
32 |
```python
|
33 |
-
from transformers import Wav2Vec2FeatureExtractor,
|
34 |
from datasets import load_dataset
|
35 |
import torch
|
36 |
|
37 |
dataset = load_dataset("hf-internal-testing/librispeech_asr_demo", "clean", split="validation")
|
38 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained('microsoft/wavlm-base-sv')
|
39 |
-
model =
|
40 |
|
41 |
# audio files are decoded on the fly
|
42 |
inputs = feature_extractor(dataset[:2]["audio"]["array"], return_tensors="pt")
|
|
|
1 |
---
|
2 |
language:
|
3 |
- en
|
|
|
4 |
tags:
|
5 |
- speech
|
6 |
---
|
|
|
29 |
# Usage
|
30 |
## Speaker Verification
|
31 |
```python
|
32 |
+
from transformers import Wav2Vec2FeatureExtractor, WavLMForXVector
|
33 |
from datasets import load_dataset
|
34 |
import torch
|
35 |
|
36 |
dataset = load_dataset("hf-internal-testing/librispeech_asr_demo", "clean", split="validation")
|
37 |
feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained('microsoft/wavlm-base-sv')
|
38 |
+
model = WavLMForXVector.from_pretrained('microsoft/wavlm-base-sv')
|
39 |
|
40 |
# audio files are decoded on the fly
|
41 |
inputs = feature_extractor(dataset[:2]["audio"]["array"], return_tensors="pt")
|