Update README.md

README.md

@@ -83,6 +83,53 @@ Base Model: meta-llama/Llama-3.1-8B-Instruct
 | Thai           | Cambodian       | 78.52      | 91.47   | 91.16   |
 | Thai           | Indonesian      | 58.99      | 78.56   | 76.40   |
 
+##
+
+```py
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+model = AutoModelForCausalLM.from_pretrained(
+    "MLP-KTLim/llama-3.1-8B-Asian-Translator",
+    torch_dtype="auto",
+    device_map="auto",
+)
+
+tokenizer = AutoTokenizer.from_pretrained(
+    "MLP-KTLim/llama-3.1-8B-Asian-Translator",
+)
+
+input_text = "안녕하세요? 아시아 언어 번역 모델 입니다."
+
+def get_input_ids(source_lang, target_lang, message):
+    assert source_lang in ["Korean", "Vietnamese", "Indonesian", "Thai", "Cambodian"]
+    assert target_lang in ["Korean", "Vietnamese", "Indonesian", "Thai", "Cambodian"]
+    
+    input_ids = tokenizer.apply_chat_template(
+        conversation=[
+            {"role": "system", "content": f"You are a useful translation AI. Please translate the sentence given in {source_lang} into {target_lang}."},
+            {"role": "user", "content": message},
+        ],
+        tokenize=True,
+        return_tensors="pt",
+        add_generation_prompt=True,
+    )
+    return input_ids
+
+input_ids = get_input_ids(
+    source_lang="Korean",
+    target_lang="Vietnamese",
+    message=input_text,
+)
+
+output = model.generate(
+    input_ids.to(model.device),
+    max_new_tokens=128,
+)
+
+print(tokenizer.decode(output[0][len(input_ids[0]):], skip_special_tokens=True))
+```
+
+
 ## Contributor
 - 원인호
 - 김민준