Update src/main.py

src/main.py

@@ -2,7 +2,7 @@ import display_gloss as dg
 import synonyms_preprocess as sp
 from NLP_Spacy_base_translator import NlpSpacyBaseTranslator 
 from flask import Flask, render_template, Response, request
-from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqGeneration
+from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqTranslation
 import torch
 import os
 
@@ -23,7 +23,7 @@ os.environ['CUDA_VISIBLE_DEVICES'] = ''
 # 번역 모델 초기화
 model_name = "Helsinki-NLP/opus-mt-ko-en"
 tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir=cache_dir)
-model = AutoModelForSeq2SeqGeneration.from_pretrained(model_name, cache_dir=cache_dir)
+model = AutoModelForSeq2SeqTranslation.from_pretrained(model_name, cache_dir=cache_dir)
 model = model.to(device)
 
 nlp, dict_docs_spacy = sp.load_spacy_values()