speechbrain
/

sepformer-wsj02mix

Source Separation

Speech Separation

Audio Source Separation

audio-source-separation

Model card Files Files and versions Community

cemsubakan commited on Mar 14, 2021

Commit

05e14ee

•

1 Parent(s): 3395766

Update README.md

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -44,9 +44,10 @@ Please notice that we encourage you to read our tutorials and learn more about
 from speechbrain.pretrained import SepformerSeparation as separator
 import torchaudio
-model = separator.from_hparams(source="speechbrain/sepformer-wsj02mix", savedir='pretrained_models/sepformer-wsj02mix')
-est_sources = model.separate_file('your_speechbrain_path/samples/audio_samples/test_mixture.wav')
 torchaudio.save("source1hat.wav", est_sources[:, :, 0].detach().cpu(), 8000)
 torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)

 from speechbrain.pretrained import SepformerSeparation as separator
 import torchaudio
+model = separator.from_hparams(source="speechbrain/sepformer-wsj02mix", savedir='sepformer-wsj02mix')
+# for custom file, change path
+est_sources = model.separate_file(path='speechbrain/sepformer-wsj02mix/test_mixture.wav')
 torchaudio.save("source1hat.wav", est_sources[:, :, 0].detach().cpu(), 8000)
 torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)