update

translate_utils.py

@@ -0,0 +1,16 @@
+from transformers import MarianMTModel, MarianTokenizer
+
+# 1) MarianMT 모델과 토크나이저 불러오기
+model_name = "Helsinki-NLP/opus-mt-ko-en"
+tokenizer = MarianTokenizer.from_pretrained(model_name)
+model = MarianMTModel.from_pretrained(model_name)
+
+# 2) 번역 함수 예시
+def translate_ko_to_en(text: str) -> str:
+    # 입력 문장을 토큰화
+    inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
+    # 모델로 추론
+    translated = model.generate(**inputs)
+    # 토큰을 다시 문장으로 디코딩
+    translation = tokenizer.batch_decode(translated, skip_special_tokens=True)
+    return translation[0]