Update README.md

README.md

@@ -10,8 +10,8 @@ license: apache-2.0
 ---
 
 ## Quick Start
-This model is upgraded version of demo_math_version. 
-Trained on 1500 row clean data
+This model is upgraded version of demo_math_model. 
+Trained on 1500 row clean data instead of raw data
 
 ### 🤗 Hugging Face Transformers
 
@@ -21,8 +21,8 @@ Here we show a code snippet to show you how to use the chat model with `transfor
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
 
-model = AutoModelForCausalLM.from_pretrained("thanghf/demo_math_model",torch_dtype=torch.bfloat16,device_map='auto')
-tokenizer = AutoTokenizer.from_pretrained("thanghf/demo_math_model")
+model = AutoModelForCausalLM.from_pretrained("thanghf/math_model_v1",torch_dtype=torch.bfloat16,device_map='auto')
+tokenizer = AutoTokenizer.from_pretrained("thanghf/math_model_v1")
 model.eval()
 streamer = TextStreamer(tokenizer)
 prompt = """Gieo hai con súc xắc cân đối và đồng chất. Xác suất để tổng số chấm trên mặt xuất hiện của hai con súc xắc bằng 7 là:"""