Update README.md

README.md

@@ -8,6 +8,43 @@ pipeline_tag: translation
 
 # Japanese to Korean translator
 
+Japanese to Korean translator model based on [EncoderDecoderModel](https://huggingface.co/docs/transformers/model_doc/encoder-decoder)([bert-japanese](https://huggingface.co/cl-tohoku/bert-base-japanese)+[kogpt2](https://github.com/SKT-AI/KoGPT2))
+
+# Usage
+
+## Inference
+
+```Python
+from transformers import(
+    EncoderDecoderModel,
+    PreTrainedTokenizerFast,
+    BertJapaneseTokenizer,
+)
+
+import torch
+
+encoder_model_name = "cl-tohoku/bert-base-japanese-v2"
+decoder_model_name = "skt/kogpt2-base-v2"
+
+src_tokenizer = BertJapaneseTokenizer.from_pretrained(encoder_model_name)
+trg_tokenizer = PreTrainedTokenizerFast.from_pretrained(decoder_model_name)
+
+model = EncoderDecoderModel.from_pretrained("sappho192/aihub-ja-ko-translator")
+
+text = "初めまして。よろしくお願いします。"
+
+def translate(text_src):
+    embeddings = src_tokenizer(text_src, return_attention_mask=False, return_token_type_ids=False, return_tensors='pt')
+    embeddings = {k: v for k, v in embeddings.items()}
+    output = model.generate(**embeddings)[0, 1:-1]
+    text_trg = trg_tokenizer.decode(output.cpu())
+    return text_trg
+
+print(translate(text))
+```
+
+# Dataset
+
 This model used datasets from 'The Open AI Dataset Project (AI-Hub, South Korea)'.  
 All data information can be accessed through 'AI-Hub ([aihub.or.kr](https://www.aihub.or.kr))'.  
 (**In order for a corporation, organization, or individual located outside of Korea to use AI data, etc., a separate agreement is required** with the performing organization and the Korea National Information Society agency(NIA). In order to export AI data, etc. outside the country, a separate agreement is required with the performing organization and the NIA. [Link](https://aihub.or.kr/intrcn/guid/usagepolicy.do?currMenu=151&topMenu=105))