Update example for ResNet model

README.md

@@ -63,16 +63,13 @@ To use the mimic-loss-trained model for enhancement, use the following simple co
 
 ```python
 import torchaudio
-from speechbrain.pretrained import SpectralMaskEnhancement
+from speechbrain.pretrained import WaveformEnhancement
 
-enhance_model = SpectralMaskEnhancement.from_hparams(
+enhance_model = WaveformEnhancement.from_hparams(
     source="speechbrain/mtl-mimic-voicebank",
     savedir="pretrained_models/mtl-mimic-voicebank",
 )
-noisy = enhance_model.load_audio("speechbrain/mtl-mimic-voicebank/example.wav")
-
-# Add batch dimension and enhance
-enhanced = enhance_model.enhance_batch(noisy.unsqueeze(0))
+enhanced = enhance_model.enhance_file("speechbrain/mtl-mimic-voicebank/example.wav")
 
 # Saving enhanced signal on disk
 torchaudio.save('enhanced.wav', enhanced.cpu(), 16000)