Update README.md

README.md

@@ -21,3 +21,24 @@ tags:
 
 - It is specialized for translating Korean medical terms into English. For usage instructions, please refer to the documentation on Transformer models and NLLB-related materials.
 
+
+```python
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+# Load model directly
+tokenizer = AutoTokenizer.from_pretrained("Woondsc/nllb-1.3B-KMA-KCD")
+model = AutoModelForSeq2SeqLM.from_pretrained("Woondsc/nllb-1.3B-KMA-KCD")
+
+# Transformer function setting
+def translate(text, model, tokenizer, target_lang="eng_Latn"):
+    inputs = tokenizer(text, return_tensors="pt")
+    inputs["forced_bos_token_id"] = tokenizer.convert_tokens_to_ids(target_lang)
+    translated_tokens = model.generate(**inputs)
+    translated_text = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
+    return translated_text
+
+# Execute example
+korean_text = "간질"
+english_translation = translate(korean_text, model, tokenizer)
+print("번역 결과:", english_translation)
+```