Update README.md

README.md

@@ -56,7 +56,7 @@ Function Calling Benchmark: https://huggingface.co/datasets/phamhai/Vietnamese-F
 | [gpt-4o-2024-08-06](https://platform.openai.com/docs/models#gpt-4o)         | ---        | 94.38    | 52.88     |
 | [arcee-ai/Arcee-VyLinh](https://huggingface.co/arcee-ai/Arcee-VyLinh)     | ~3B        |     |     | 
 
-<p align="left"> Table 1. [Vietnamese Function Calling Benchmark](https://huggingface.co/datasets/phamhai/Vietnamese-Function-Calling-Test) </p>
+<p align="left"> Table 1. Vietnamese Function Calling Benchmark </p>
 
 <h2> Run the model </h2>
 
@@ -76,7 +76,7 @@ model = AutoModelForCausalLM.from_pretrained(model_path)
 messages =  [
     {"role": "system", "content": "Bạn là một người bạn gái xinh đẹp. Tên của bạn là Vivi. Hãy luôn xưng là Vivi, gọi người nói là anh và trả lời luôn bắt đầu bằng cụm từ Dạ thưa anh yêu của em."},
     {"role": "user", "content": "xin chào em"}]
-tokenized_chat = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=False, return_tensors="pt")
+tokenized_chat = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt")
 
 outputs = model.generate(tokenized_chat, max_new_tokens=128) 
 print(tokenizer.decode(outputs[0]))