Update app.py

app.py

@@ -19,12 +19,6 @@ import gradio as gr
 from scipy.io import wavfile
 from scipy.io.wavfile import write
 
-from speechbrain.pretrained import SpectralMaskEnhancement
-
-enhance_model = SpectralMaskEnhancement.from_hparams(
-    source="speechbrain/metricgan-plus-voicebank",
-    savedir="pretrained_models/metricgan-plus-voicebank",
-)
 
 knn_vc = torch.hub.load('bshall/knn-vc', 'knn_vc', prematched=True, trust_repo=True, pretrained=True, device='cpu')
 
@@ -51,12 +45,7 @@ def voice_change(audio_in, audio_ref):
     matching_set = knn_vc.get_matching_set(["./audio_ref.wav"])
     out_wav = knn_vc.match(query_seq, matching_set, topk=4)
     torchaudio.save('output.wav', out_wav[None], 16000)
-    noisy = enhance_model.load_audio(
-        'output.wav'
-    ).unsqueeze(0)
-    enhanced = enhance_model.enhance_batch(noisy, lengths=torch.tensor([1.]))
-    torchaudio.save('enhanced.wav', enhanced.cpu(), 16000)
-    return 'enhanced.wav'
+    return 'output.wav'
 
 
 def tts(text, model, voice, api_key):