speechbrainteam
commited on
Commit
·
59bcf9c
1
Parent(s):
720f1ce
Update README.md
Browse files
README.md
CHANGED
@@ -76,7 +76,7 @@ The code will automatically normalize your audio (i.e., resampling + mono channe
|
|
76 |
First of all, please install tranformers and SpeechBrain with the following command:
|
77 |
|
78 |
```
|
79 |
-
pip install speechbrain
|
80 |
```
|
81 |
|
82 |
Please notice that we encourage you to read our tutorials and learn more about
|
@@ -91,17 +91,17 @@ import torch
|
|
91 |
|
92 |
enh_model = Separator.from_hparams(
|
93 |
source="speechbrain/noisy-whisper-resucespeech",
|
94 |
-
savedir='pretrained_models/noisy-whisper-
|
95 |
hparams_file="enhance.yaml"
|
96 |
)
|
97 |
asr_model = WhisperASR.from_hparams(
|
98 |
source="speechbrain/noisy-whisper-resucespeech",
|
99 |
-
savedir="pretrained_models/noisy-whisper-
|
100 |
hparams_file="asr.yaml"
|
101 |
)
|
102 |
|
103 |
# For custom file, change the path accordingly
|
104 |
-
est_sources = enh_model.separate_file(path='example_rescuespeech16k.wav')
|
105 |
pred_words, _ = asr_model(est_sources[:, :, 0], torch.tensor([1.0]))
|
106 |
```
|
107 |
### Inference on GPU
|
|
|
76 |
First of all, please install tranformers and SpeechBrain with the following command:
|
77 |
|
78 |
```
|
79 |
+
pip install speechbrain
|
80 |
```
|
81 |
|
82 |
Please notice that we encourage you to read our tutorials and learn more about
|
|
|
91 |
|
92 |
enh_model = Separator.from_hparams(
|
93 |
source="speechbrain/noisy-whisper-resucespeech",
|
94 |
+
savedir='pretrained_models/noisy-whisper-rescuespeech',
|
95 |
hparams_file="enhance.yaml"
|
96 |
)
|
97 |
asr_model = WhisperASR.from_hparams(
|
98 |
source="speechbrain/noisy-whisper-resucespeech",
|
99 |
+
savedir="pretrained_models/noisy-whisper-rescuespeech",
|
100 |
hparams_file="asr.yaml"
|
101 |
)
|
102 |
|
103 |
# For custom file, change the path accordingly
|
104 |
+
est_sources = enh_model.separate_file(path='speechbrain/noisy-whisper-resucespeech/example_rescuespeech16k.wav')
|
105 |
pred_words, _ = asr_model(est_sources[:, :, 0], torch.tensor([1.0]))
|
106 |
```
|
107 |
### Inference on GPU
|