Upload wheel

app.py

@@ -7,8 +7,11 @@ import torch
 # install packages for mamba
 def install():
     print("Install personal packages", flush=True)
-    subprocess.run(shlex.split("pip install mamba-ssm==1.2.0.post1"))
-    #subprocess.run(shlex.split("pip install causal_conv1d-1.0.0-cp310-cp310-linux_x86_64.whl"))
+    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?download=true"))
+    subprocess.run(shlex.split("wget https://huggingface.co/spaces/OpenGVLab/VideoMamba/resolve/main/mamba_ssm-1.0.1-cp310-cp310-linux_x86_64.whl?download=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"))
     #subprocess.run(shlex.split("pip install ./mamba_install"))
 
 install()