Update README.md
Browse files
README.md
CHANGED
@@ -39,6 +39,7 @@ To use the mimic-loss-trained model for enhancement, use the following simple co
|
|
39 |
|
40 |
```python
|
41 |
import torch
|
|
|
42 |
from speechbrain.pretrained import SpectralMaskEnhancement
|
43 |
|
44 |
enhance_model = SpectralMaskEnhancement.from_hparams(
|
@@ -52,7 +53,10 @@ noisy = enhance_model.load_audio(
|
|
52 |
).unsqueeze(0)
|
53 |
|
54 |
# Add relative length tensor
|
55 |
-
enhance_model.enhance_batch(noisy, lengths=torch.tensor([1.]))
|
|
|
|
|
|
|
56 |
```
|
57 |
|
58 |
## Referencing MetricGAN+
|
@@ -79,6 +83,6 @@ title = {SpeechBrain},
|
|
79 |
year = {2021},
|
80 |
publisher = {GitHub},
|
81 |
journal = {GitHub repository},
|
82 |
-
howpublished = {
|
83 |
}
|
84 |
```
|
|
|
39 |
|
40 |
```python
|
41 |
import torch
|
42 |
+
import torchaudio
|
43 |
from speechbrain.pretrained import SpectralMaskEnhancement
|
44 |
|
45 |
enhance_model = SpectralMaskEnhancement.from_hparams(
|
|
|
53 |
).unsqueeze(0)
|
54 |
|
55 |
# Add relative length tensor
|
56 |
+
enhanced = enhance_model.enhance_batch(noisy, lengths=torch.tensor([1.]))
|
57 |
+
|
58 |
+
# Saving enhanced signal on disk
|
59 |
+
torchaudio.save('enhanced.wav', enhanced, 16000)
|
60 |
```
|
61 |
|
62 |
## Referencing MetricGAN+
|
|
|
83 |
year = {2021},
|
84 |
publisher = {GitHub},
|
85 |
journal = {GitHub repository},
|
86 |
+
howpublished = {\\url{https://github.com/speechbrain/speechbrain}},
|
87 |
}
|
88 |
```
|