Upload wheel

app.py

@@ -4,16 +4,11 @@ import spaces
 import torch
 
 # install packages for mamba
-def install():
-    print("Install personal packages", flush=True)
-    print("Refer to OpenGVLab / VideoMamba")
-    subprocess.run(shlex.split("wget https://huggingface.co/spaces/OpenGVLab/VideoMamba/resolve/main/causal_conv1d-1.0.0-cp310-cp310-linux_x86_64.whl"))
-    subprocess.run(shlex.split("wget https://huggingface.co/spaces/OpenGVLab/VideoMamba/resolve/main/mamba_ssm-1.0.1-cp310-cp310-linux_x86_64.whl"))
-    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"))
-    #subprocess.run(shlex.split("pip install ./mamba_install"))
-
-install()
+def install_mamba():
+    subprocess.run(shlex.split("pip install https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.4.0/causal_conv1d-1.4.0+cu122torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl"))
+    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"))
+
+install_mamba()
 
 
 import gradio as gr