Update app.py

app.py

@@ -19,15 +19,15 @@ from refiners.solutions import BoxSegmenter
 from transformers import GroundingDinoForObjectDetection, GroundingDinoProcessor
 from diffusers import FluxPipeline
 # 상단에 import 추가
-# 상단에 import 추가
-from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqGeneration
+from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqTranslation
 
 # 번역 모델 초기화
 model_name = "Helsinki-NLP/opus-mt-ko-en"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqGeneration.from_pretrained(model_name)
+model = AutoModelForSeq2SeqTranslation.from_pretrained(model_name)
 translator = pipeline("translation", model=model, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)
 
+
 def translate_to_english(text: str) -> str:
     """한글 텍스트를 영어로 번역"""
     try: