Hugging Face's logo

speechbrain
/
sepformer-wsj02mix

Audio-to-Audio
speechbrain
English
Source Separation
Speech Separation
Audio Source Separation
WSJ02Mix
SepFormer
Transformer
audio-source-separation
Model card Files Community
1
cemsubakan commited on
Commit
05e14ee
·
1 Parent(s): 3395766

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -44,9 +44,10 @@ Please notice that we encourage you to read our tutorials and learn more about
44
  from speechbrain.pretrained import SepformerSeparation as separator
45
  import torchaudio
46
 
47
- model = separator.from_hparams(source="speechbrain/sepformer-wsj02mix", savedir='pretrained_models/sepformer-wsj02mix')
48
 
49
- est_sources = model.separate_file('your_speechbrain_path/samples/audio_samples/test_mixture.wav')
 
50
 
51
  torchaudio.save("source1hat.wav", est_sources[:, :, 0].detach().cpu(), 8000)
52
  torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)
 
44
  from speechbrain.pretrained import SepformerSeparation as separator
45
  import torchaudio
46
 
47
+ model = separator.from_hparams(source="speechbrain/sepformer-wsj02mix", savedir='sepformer-wsj02mix')
48
 
49
+ # for custom file, change path
50
+ est_sources = model.separate_file(path='speechbrain/sepformer-wsj02mix/test_mixture.wav')
51
 
52
  torchaudio.save("source1hat.wav", est_sources[:, :, 0].detach().cpu(), 8000)
53
  torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)