fix readme

README.md

@@ -13,6 +13,8 @@ Dựa trên [Qwen2.5-14B-Instruct](https://huggingface.co/Qwen/Qwen2.5-14B-Instr
 Mô hình dùng để tính toán độ tương đồng giữa văn bản tiếng Trung và Việt
 
 ```python
+import torch
+import torch.nn.functional as F
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers.generation import GenerationConfig
 
@@ -21,6 +23,9 @@ tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_path, device_map='auto', trust_remote_code=True).eval()
 model.generation_config = GenerationConfig.from_pretrained(model_path, trust_remote_code=True)
 
+yes_id = tokenizer.encode('Yes')[0]
+no_id = tokenizer.encode('No')[0]
+
 zh_text = '''听完了倒计时，封不觉很快便进入了游戏的世界中。他置身在一个和电梯十分相像的环境里，但门的旁边没有楼层按钮。右手边的墙壁上是个三十二寸的触摸屏。正对门的那一整面墙是一块完整的镜面，可以将玩家从头照到脚。
 封不觉看了看镜子中自己的形象，此刻他成了个超高像素的三维CG人物。身上的服装变成了黑色的长袖T恤和长裤，没什么特别之处。游戏人物的脸和现实中自己的长相几乎一样，一米八不到，乱糟糟的头发，面部线条略显阴柔。
 在第四代光脑技术的支持下，神经连接游戏中的玩家可以获得足以乱真的五感体验，但游戏画面依然局限于CG影像，而无法让玩家载入到类似真人实景拍摄效果下的环境。'''