Hugging Face's logo

Spaces:
hieupt
/
Sepformer_denosie_audio
Running

App Files Community
hieupt commited on
Commit
a7cac90
•
1 Parent(s): 314acb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,7 +2,8 @@ import gradio as gr
2
  from speechbrain.pretrained import SepformerSeparation as separator
3
  import torchaudio
4
 
5
- model = separator.from_hparams(source="speechbrain/sepformer-wham-enhancement", savedir='pretrained_models/sepformer-wham-enhancement')
 
6
 
7
  def predict_song(audio_path):
8
  est_sources = model.separate_file(path=audio_path)
 
2
  from speechbrain.pretrained import SepformerSeparation as separator
3
  import torchaudio
4
 
5
+ # model = separator.from_hparams(source="speechbrain/sepformer-wham-enhancement", savedir='pretrained_models/sepformer-wham-enhancement')
6
+ model = separator.from_hparams(source="speechbrain/sepformer-dns4-16k-enhancement", savedir='pretrained_models/sepformer-dns4-16k-enhancement')
7
 
8
  def predict_song(audio_path):
9
  est_sources = model.separate_file(path=audio_path)