Update app.py

app.py

@@ -7,7 +7,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 import torch
 import os
 import bitnet
-os.system('pip install mamba-ssm causal-conv1d>=1.2.0')
 key =  os.environ.get("key")
 from huggingface_hub import login 
 login(key)
@@ -21,7 +20,7 @@ nf4_config = BitsAndBytesConfig(
    bnb_4bit_use_double_quant=True,
    bnb_4bit_compute_dtype=torch.bfloat16
 )
-model_id = "ai21labs/Jamba-v0.1"
+model_id = "IEITYuan/Yuan2-M32-hf"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id,
                                              # load_in_8bit=True,