Upload

app.py

@@ -10,6 +10,8 @@ def install_mamba():
     subprocess.run(shlex.split("pip install https://github.com/state-spaces/mamba/releases/download/v2.2.2/mamba_ssm-2.2.2+cu122torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl"))
     subprocess.run(shlex.split("pip install numpy==1.26.4"))
 
+    subprocess.run(shlex.split("ls"))
+
 
 install_mamba()
 
@@ -24,10 +26,14 @@ from models.generator import SEMamba
 from models.pcs400   import cal_pcs
 
 # download model files from your HF repo
-ckpt  = hf_hub_download("rc19477/Speech_Enhancement_Mamba",
-                        "ckpts/SEMamba_advanced.pth")
-cfg_f = hf_hub_download("rc19477/Speech_Enhancement_Mamba",
-                        "recipes/SEMamba_advanced.yaml")
+#ckpt  = hf_hub_download("rc19477/Speech_Enhancement_Mamba",
+#                        "ckpts/SEMamba_advanced.pth")
+#cfg_f = hf_hub_download("rc19477/Speech_Enhancement_Mamba",
+#                        "recipes/SEMamba_advanced.yaml")
+
+ckpt = "ckpts/SEMamba_advanced.pth"
+cfg_f = "recipes/SEMamba_advanced.yaml"
+
 
 # load config
 with open(cfg_f) as f: