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])