| from speechbrain.pretrained import SpectralMaskEnhancement | |
| import torchaudio | |
| model = SpectralMaskEnhancement.from_hparams(source="speechbrain/metricgan-plus-voicebank") | |
| def remove_noise(input_path, output_path): | |
| enhanced = model.enhance_file(input_path) | |
| torchaudio.save(output_path, enhanced[0], enhanced[1]) | |