Shokoufehhh commited on
Commit
7cfdfe4
1 Parent(s): 700885d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import torch
2
  import gradio as gr
3
- from model import SGMSE # Adjust according to your model's definition
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"
 
1
  import torch
2
  import gradio as gr
3
+ 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"