Update README.md

README.md

@@ -1,38 +1,42 @@
 ---
 language:
-- ko
-- ja
-- zh
+  - ko
+  - ja
+  - zh
+  - en
 license: apache-2.0
 library_name: optimum
 tags:
-- translation
-- m2m100
-- korean
-- japanese
-- chinese
-- k-tourism
-- onnx
+  - translation
+  - m2m100
+  - korean
+  - japanese
+  - chinese
+  - english
+  - k-tourism
+  - onnx
 pipeline_tag: translation
 base_model: facebook/m2m100_1.2B
 datasets:
-- custom-k-tourism-corpus
+  - custom-k-tourism-corpus
 ---
 
 # M2M100 Korean Tourism Translator (ONNX)
 
-이 모델은 `facebook/m2m100_1.2B` 모델을 **한국 관광** 관련 데이터로 파인튜닝(Fine-tuning)한 후, 추론 속도 향상을 위해 **ONNX (Open Neural Network Exchange)** 형식으로 변환한 번역 모델입니다.
+이 모델은 `facebook/m2m100_1.2B` 모델을 한국 관광 관련 데이터로 파인튜닝(Fine-tuning)한 후, 추론 속도 향상을 위해 ONNX (Open Neural Network Exchange) 형식으로 변환한 번역 모델입니다.
 
-주요 기능은 한국어(ko)로 된 관광 관련 텍스트를 일본어(ja)와 중국어(zh)로 번역하는 것입니다.
+주요 기능은 한국어(ko)와 영어(en), 일본어(ja), 중국어(zh) 간의 양방향 번역을 수행하는 것입니다.
 
-- **Base Model:** [facebook/m2m100_1.2B](https://huggingface.co/facebook/m2m100_1.2B)
-- **Specialization:** Korean Tourism Domain (한국 관광 특화)
-- **Target Languages:** Japanese (ja), Chinese (zh)
-- **Format:** ONNX (Optimized for fast CPU/GPU inference)
+- **Base Model**: `facebook/m2m100_1.2B`
+- **Specialization**: Korean Tourism Domain (한국 관광 특화)
+- **Target Languages**: English (en), Japanese (ja), Chinese (zh)
+- **Format**: ONNX (Optimized for fast CPU/GPU inference)
 
 ## Model Description
 
-M2M100은 별도의 언어 지정 없이 100개 언어 간의 번역을 수행할 수 있는 다국어 번역 모델입니다. 이 모델은 그 중에서도 특히 한국 관광 분야의 용어와 문체에 대한 이해도를 높이기 위해, 관련 데이터셋으로 추가 학습을 진행했습니다.
+M2M100은 별도의 언어 지정 없이 100개 언어 간의 번역을 수행할 수 있는 다국어 번역 모델입니다. 이 모델은 그 중에서도 특히 한국 관광 분야의 용어와 문체에 대한 이해도를 높이기 위해, 관련 데이터셋으로 추가 학습을 진행했습니다. 
+
+특히, 한국의 특정 지명(예: 경복궁, 명동)이나 음식 이름(예: 비빔밥, 떡볶이) 등 고유명사에 대한 3개 언어(영어, 일본어, 중국어) 번역 정확도를 높이는 데 중점을 두었습니다.
 
 ONNX 변환을 통해 기존 PyTorch 모델보다 가볍고 빠르게 작동하므로, FastAPI 등을 이용한 API 서버 배포에 매우 적합합니다.
 
@@ -83,24 +87,44 @@ result_zh = translator(
 )
 print(f"Korean to Chinese: {result_zh[0]['translation_text']}")
 
+# 한국어 -> 영어 번역
+result_en = translator(
+    korean_text,
+    src_lang="ko",
+    tgt_lang="en"
+)
+print(f"Korean to English: {result_en[0]['translation_text']}")
+
+# 일본어 -> 한국어 번역
+japanese_text = "景福宮の夜間開場入場券はどこで購入できますか。"
+result_ko_from_ja = translator(
+    japanese_text,
+    src_lang="ja",
+    tgt_lang="ko"
+)
+print(f"Japanese to Korean: {result_ko_from_ja[0]['translation_text']}")
+
 # --- 출력 결과 예시 ---
 # Korean to Japanese: 景福宮の夜間開場入場券はどこで購入できますか。
 # Korean to Chinese: 景福宫夜间开放门票在哪里购买?
+# Korean to English: Where can I buy tickets for the Gyeongbok Palace night opening?
+# Japanese to Korean: 경복궁 야간 개장 입장권은 어디에서 구입합니까?
 ```
 
 ## Model Details
 
 ### Fine-tuning
-
-- **Base Model:** `facebook/m2m100_1.2B`
-- **Training Data:** 자체적으로 수집 및 정제한 한국 관광 관련 문장 쌍 데이터셋 (K-Tourism Corpus)으로 파인튜닝을 진행했습니다. 이 데이터셋에는 관광지 정보, 교통, 숙박, 음식, 쇼핑 등 다양한 카테고리의 내용이 포함되어 있습니다.
-- **Objective:** 일반적인 번역 모델이 어색하게 번역할 수 있는 고유명사나 특정 상황(티켓 예매, 길 찾기 등)에 대한 번역 품질을 높이는 것을 목표로 했습니다.
+- **Base Model**: `facebook/m2m100_1.2B`
+- **Training Data**: 자체적으로 수집 및 정제한 한국 관광 관련 문장 쌍 데이터셋 (K-Tourism Corpus)으로 파인튜닝을 진행했습니다. 이 데이터셋에는 관광지 정보, 음식, 행사 등 다양한 카테고리의 내용이 포함되어 있습니다.
+- **Objective**: 일반적인 번역 모델이 어색하게 번역할 수 있는 한국 관광 관련 고유명사(지명, 음식 이름 등)에 대한 3개 언어 번역 품질을 높이는 것을 목표로 했습니다.
 
 ### ONNX Conversion
-
-- **Performance:** PyTorch 모델을 ONNX로 변환하여 양자화(quantization) 및 최적화를 거쳤습니다. 이를 통해 CPU 환경에서도 더 빠른 추론 속도를 제공하며, 이는 실���간 번역 API 서비스에 큰 이점입니다.
-- **Compatibility:** ONNX Runtime은 다양한 하드웨어 및 플랫폼을 지원하여 모델 배포의 유연성을 높여줍니다.
+- **Performance**: PyTorch 모델을 ONNX로 변환하여 양자화(quantization) 및 최적화를 거쳤습니다. 이를 통해 CPU 환경에서도 더 빠른 추론 속도를 제공하며, 이는 실시간 번역 API 서비스에 큰 이점입니다.
+- **Compatibility**: ONNX Runtime은 다양한 하드웨어 및 플랫폼을 지원하여 모델 배포의 유연성을 높여줍니다.
 
 ## Deployment
 
 이 ONNX 모델은 FastAPI와 Docker를 사용하여 API 서버로 쉽게 배포할 수 있도록 설계되었습니다. 자세한 배포 방법은 관련 프로젝트의 `Dockerfile`과 `app.py`를 참고하세요.
+
+### Docker Hub Image
+- **[Repository](https://hub.docker.com/repository/docker/pokqok/m2m100-k-tourism-ko-ja-zh-onnx/general)**: