roychao19477 commited on
Commit
c450417
·
1 Parent(s): 746a4fa

Update mamba issue

Browse files
Files changed (1) hide show
  1. app.py +10 -9
app.py CHANGED
@@ -1,12 +1,3 @@
1
- import gradio as gr
2
- import torch
3
- import yaml
4
- import librosa
5
- from huggingface_hub import hf_hub_download
6
- from models.stfts import mag_phase_stft, mag_phase_istft
7
- from models.generator import SEMamba
8
- from models.pcs400 import cal_pcs
9
-
10
  import shlex
11
  import subprocess
12
  import spaces
@@ -21,6 +12,16 @@ def install():
21
 
22
  install()
23
 
 
 
 
 
 
 
 
 
 
 
24
  # download model files from your HF repo
25
  ckpt = hf_hub_download("rc19477/Speech_Enhancement_Mamba",
26
  "ckpts/SEMamba_advanced.pth")
 
 
 
 
 
 
 
 
 
 
1
  import shlex
2
  import subprocess
3
  import spaces
 
12
 
13
  install()
14
 
15
+
16
+ import gradio as gr
17
+ import torch
18
+ import yaml
19
+ import librosa
20
+ from huggingface_hub import hf_hub_download
21
+ from models.stfts import mag_phase_stft, mag_phase_istft
22
+ from models.generator import SEMamba
23
+ from models.pcs400 import cal_pcs
24
+
25
  # download model files from your HF repo
26
  ckpt = hf_hub_download("rc19477/Speech_Enhancement_Mamba",
27
  "ckpts/SEMamba_advanced.pth")