switch to voice enhancement

app.py

@@ -2,13 +2,12 @@ from speechbrain.inference.separation import SepformerSeparation as separator
 import torchaudio
 import gradio as gr
 
-model = separator.from_hparams(source="speechbrain/sepformer-wsj02mix", savedir='pretrained_models/sepformer-wsj02mix')
+model = separator.from_hparams(source="speechbrain/sepformer-whamr-enhancement", savedir='pretrained_models/sepformer-whamr-enhancement')
 
 def speechbrain(aud):
   est_sources = model.separate_file(path=aud)
   torchaudio.save("source1hat.wav", est_sources[:, :, 0].detach().cpu(), 8000)
-  torchaudio.save("source2hat.wav", est_sources[:, :, 1].detach().cpu(), 8000)
-  return "source1hat.wav", "source2hat.wav"
+  return "source1hat.wav"
 
 
 title = "Speech Seperation"
@@ -22,8 +21,7 @@ demo = gr.Interface(
   fn=speechbrain,
   inputs=gr.Audio(type="filepath"),
   outputs=[
-    gr.Audio(label="Output Audio One", type="filepath"),
-    gr.Audio(label="Output Audio Two", type="filepath")
+    gr.Audio(label="Output Audio", type="filepath")
   ],
   title=title,
   description=description,

requirements.txt

@@ -1,3 +1,3 @@
 speechbrain
 torchaudio
-gradio==3.3.1
+gradio