Update app.py

app.py

@@ -19,12 +19,13 @@ from refiners.solutions import BoxSegmenter
 from transformers import GroundingDinoForObjectDetection, GroundingDinoProcessor
 from diffusers import FluxPipeline
 # 상단에 import 추가
-from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqTranslation
+# 상단에 import 추가
+from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 
 # 번역 모델 초기화
 model_name = "Helsinki-NLP/opus-mt-ko-en"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqTranslation.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 translator = pipeline("translation", model=model, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)