Update model_simple.py

model_simple.py

@@ -8,8 +8,6 @@ from typing import Optional, Dict
 import numpy as np
 from datetime import datetime
 from dataclasses import dataclass
-from transformers.trainer_seq2seq import Seq2SeqTrainer
-from transformers.training_args_seq2seq import Seq2SeqTrainingArguments
 from torch.nn.functional import scaled_dot_product_attention
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")