Update mamba issue

app.py

@@ -7,6 +7,20 @@ from models.stfts    import mag_phase_stft, mag_phase_istft
 from models.generator import SEMamba
 from models.pcs400   import cal_pcs
 
+import shlex
+import subprocess
+import spaces
+import torch
+
+
+# install packages for mamba
+def install():
+    print("Install personal packages", flush=True)
+    subprocess.run(shlex.split("pip install causal_conv1d-1.0.0-cp310-cp310-linux_x86_64.whl"))
+    subprocess.run(shlex.split("pip install mamba_ssm-1.0.1-cp310-cp310-linux_x86_64.whl"))
+
+install()
+
 # download model files from your HF repo
 ckpt  = hf_hub_download("rc19477/Speech_Enhancement_Mamba",
                         "ckpts/SEMamba_advanced.pth")

requirements.txt

@@ -10,5 +10,4 @@ numpy==1.26.4
 torch==2.2.2
 torchaudio==2.2.2
 numpy==1.26.4
--e /app/mamba_ssm
 numpy==1.26.4