Spaces:
Runtime error
Runtime error
Shokoufehhh
commited on
Commit
•
9424430
1
Parent(s):
db47006
Update app.py
Browse files
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
|