Audio-to-Audio
speechbrain
English
speech-enhancement
PyTorch
Mirco commited on
Commit
decfc76
·
1 Parent(s): e722b6f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -2
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 = {\url{https://github.com/speechbrain/speechbrain}},
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
  ```