Update README.md

README.md

@@ -14,11 +14,11 @@ model = AutoModelForCausalLM.from_pretrained(
     torch_dtype="auto",
     device_map="auto"
 )
-tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-14B-Chat-GPTQ-Int4")
+tokenizer = AutoTokenizer.from_pretrained("vilm/VinaLlama2-14B-GPTQ-Int4")
 
-prompt = "Give me a short introduction to large language model."
+prompt = "Việt Nam có bao nhiêu tỉnh thành?"
 messages = [
-    {"role": "system", "content": "You are a helpful assistant."},
+    {"role": "system", "content": "Bạn là một trợ lí AI hữu ích."},
     {"role": "user", "content": prompt}
 ]
 text = tokenizer.apply_chat_template(