Update README.md

change chat template!!!

README.md

@@ -92,11 +92,20 @@ source_text = "Hello, how are you?"
 source_lang = "en"
 target_lang = "ko"  # or "uz" for Uzbek
 
-input_text = f"Translate from {source_lang} to {target_lang}: {source_text}"
-input_ids = tokenizer(input_text, return_tensors="pt").input_ids
+messages = [
+        {"role": "system", "content": f"""Translate {input_lang} to {output_lang} word by word correctly."""},
+        {"role": "user", "content": f"""{source_text}"""},
+    ]
+# Apply chat template
+input_ids = tokenizer.apply_chat_template(
+        messages,
+        add_generation_prompt=True,
+        return_tensors="pt"
+    ).to('cuda')
 
 outputs = model.generate(input_ids, max_length=100)
-translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+response = outputs[0][input_ids.shape[-1]:]
+translated_text = tokenizer.decode(response, skip_special_tokens=True)
 print(translated_text)
 ```
 ## Performance