Update README.md

README.md

@@ -7,6 +7,11 @@ base_model:
 - Qwen/Qwen2.5-7B
 pipeline_tag: text-generation
 library_name: transformers
+datasets:
+- airesearch/WangchanX-Legal-ThaiCCL-RAG
+- airesearch/WangchanThaiInstruct
+- pythainlp/han-instruct-dataset-v4.0
+- airesearch/concat_six_dataset_th_en
 ---
 
 <img src="./Tsunami.webp" alt="Tsunami Model" width="800" style="margin-left:'auto' margin-right:'auto' display:'block'"/>
@@ -42,8 +47,9 @@ This model uses `ChatML` prompt template:
 ```python
 
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
 
-model_name = "Tsunami/Tsunami-0.5-7B-Instruct"
+model_name = "Tsunami-th/Tsunami-0.5-7B-Instruct"
 
 model = AutoModelForCausalLM.from_pretrained(
     model_name,