Update README.md

README.md

@@ -73,13 +73,13 @@ from dartrs.dartrs import DartTokenizer
 from dartrs.utils import get_generation_config
 from dartrs.v2 import (
     compose_prompt,
-    MixtralModel,
+    MistralModel,
     V2Model,
 )
 import time
 import os
 MODEL_NAME = "p1atdev/dart-v2-base"
-model = MixtralModel.from_pretrained(MODEL_NAME)
+model = MistralModel.from_pretrained(MODEL_NAME)
 tokenizer = DartTokenizer.from_pretrained(MODEL_NAME)
 config = get_generation_config(
     prompt=compose_prompt(