Update app.py

app.py

@@ -2,6 +2,14 @@ import subprocess
 import sys
 import spaces
 
+# install packages for mamba
+def install():
+    print("Install personal packages", flush=True)
+    subprocess.run(shlex.split("pip install https://github.com/Dao-AILab/causal-conv1d/archive/refs/tags/v1.4.0.tar.gz"))
+    subprocess.run(shlex.split("pip install https://github.com/state-spaces/mamba/archive/refs/tags/v2.2.2.tar.gz"))
+
+install()
+
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 import gradio as gr