Update README.md

README.md

@@ -81,7 +81,7 @@ from dartrs.dartrs import DartTokenizer
 from dartrs.utils import get_generation_config
 from dartrs.v2 import (
     compose_prompt,
-    MixtralModel,
+    MistralModel,
     V2Model,
 )
 import time
@@ -89,7 +89,7 @@ import os
 
 MODEL_NAME = "p1atdev/dart-v2-sft"
 
-model = MixtralModel.from_pretrained(MODEL_NAME)
+model = MistralModel.from_pretrained(MODEL_NAME)
 tokenizer = DartTokenizer.from_pretrained(MODEL_NAME)
 
 config = get_generation_config(