lyimo commited on
Commit
251f2af
·
verified ·
1 Parent(s): b5a34d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,11 +3,12 @@ import torch
3
  import torchaudio
4
  from speechbrain.inference.enhancement import SpectralMaskEnhancement
5
  from speechbrain.inference.separation import SepformerSeparation as separator
 
6
 
7
  # Load the enhancement model
8
  model = separator.from_hparams(
9
  source="speechbrain/sepformer-dns4-16k-enhancement",
10
- savedir='pretrained_models/sepformer-dns4-16k-enhancement'
11
  )
12
 
13
  # Define the enhancement function
 
3
  import torchaudio
4
  from speechbrain.inference.enhancement import SpectralMaskEnhancement
5
  from speechbrain.inference.separation import SepformerSeparation as separator
6
+ import torchaudio
7
 
8
  # Load the enhancement model
9
  model = separator.from_hparams(
10
  source="speechbrain/sepformer-dns4-16k-enhancement",
11
+ savedir='pretrained_models/sepformer-dns4-16k-enhancement'
12
  )
13
 
14
  # Define the enhancement function