pplantinga
commited on
Commit
•
85e2e23
1
Parent(s):
93348e5
Add Pretrained example
Browse files
README.md
CHANGED
@@ -50,6 +50,20 @@ pip install speechbrain
|
|
50 |
Please notice that we encourage you to read our tutorials and learn more about
|
51 |
[SpeechBrain](https://speechbrain.github.io).
|
52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
53 |
## Referencing Mimic Loss
|
54 |
|
55 |
If you find mimic loss useful, please cite:
|
@@ -76,4 +90,4 @@ publisher = {GitHub},
|
|
76 |
journal = {GitHub repository},
|
77 |
howpublished = {\url{https://github.com/speechbrain/speechbrain}},
|
78 |
}
|
79 |
-
```
|
|
|
50 |
Please notice that we encourage you to read our tutorials and learn more about
|
51 |
[SpeechBrain](https://speechbrain.github.io).
|
52 |
|
53 |
+
## Pretrained Usage
|
54 |
+
|
55 |
+
To use the mimic-loss-trained model for enhancement, use the following simple code:
|
56 |
+
|
57 |
+
```python
|
58 |
+
from speechbrain.pretrained import SpectralMaskEnhancement
|
59 |
+
|
60 |
+
enhance_model = SpectralMaskEnhancement.from_hparams(
|
61 |
+
source="speechbrain/mtl-mimic-voicebank",
|
62 |
+
savedir="pretrained_models/mtl-mimic-voicebank",
|
63 |
+
)
|
64 |
+
enhance_model.enhance_file("/path/to/file.wav")
|
65 |
+
```
|
66 |
+
|
67 |
## Referencing Mimic Loss
|
68 |
|
69 |
If you find mimic loss useful, please cite:
|
|
|
90 |
journal = {GitHub repository},
|
91 |
howpublished = {\url{https://github.com/speechbrain/speechbrain}},
|
92 |
}
|
93 |
+
```
|