Shokoufehhh commited on
Commit
9424430
1 Parent(s): db47006

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,9 +4,9 @@ from sgmse.model import ScoreModel
4
 
5
  # Load your model
6
  model_path = "https://huggingface.co/sp-uhh/speech-enhancement-sgmse/resolve/main/pretrained_checkpoints/speech_enhancement/train_vb_29nqe0uh_epoch%3D115.ckpt"
7
- model = SGMSE() # Initialize your model class
8
  model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
9
- model.eval() # Set the model to evaluation mode
10
 
11
  def enhance_audio(input_audio):
12
  import torchaudio
 
4
 
5
  # Load your model
6
  model_path = "https://huggingface.co/sp-uhh/speech-enhancement-sgmse/resolve/main/pretrained_checkpoints/speech_enhancement/train_vb_29nqe0uh_epoch%3D115.ckpt"
7
+ #model = SGMSE() # Initialize your model class
8
  model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
9
+ #model.eval() # Set the model to evaluation mode
10
 
11
  def enhance_audio(input_audio):
12
  import torchaudio