Shokoufeh commited on
Commit
db47006
2 Parent(s): 0504c9a 7cfdfe4

Merge branch 'main' of https://huggingface.co/spaces/sp-uhh/using_torch_hub_2

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"