Update app.py

app.py

@@ -10,7 +10,7 @@ subprocess.run('pip install mistral_inference mamba-ssm --no-build-isolation', e
 subprocess.run('pip install causal-conv1d --no-build-isolation', env={'CAUSAL_CONV1D_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 # Import after installation
-from mistral_inference.transformer import Transformer
+from mistral_inference.mamba import Mamba
 from mistral_inference.generate import generate
 from mistral_common.tokens.tokenizers.mistral import MistralTokenizer
 from mistral_common.protocol.instruct.messages import UserMessage, AssistantMessage
@@ -28,7 +28,7 @@ MODEL_PATH = str(mistral_models_path)
 
 # Load model and tokenizer
 tokenizer = MistralTokenizer.from_file(os.path.join(MODEL_PATH, "tokenizer.model.v3"))
-model = Transformer.from_folder(MODEL_PATH)
+model = Mamba.from_folder(MODEL_PATH)
 
 
 @spaces.GPU()