Update app.py

app.py

@@ -5,7 +5,7 @@ import os, torch
 from datasets import load_dataset
 from huggingface_hub import HfApi, login
 from peft import LoraConfig, TaskType, get_peft_model
-from transformers import AutoModelForCausalLM, AutoTokenizer, Seq2SeqTrainer, Seq2SeqTrainingArguments, Trainer, pipeline
+from transformers import AutoModelForCausalLM, AutoTokenizer, Seq2SeqTrainer, Seq2SeqTrainingArguments, pipeline
 
 ACTION_1 = "Prompt base model"
 ACTION_2 = "Fine-tune base model"
@@ -111,7 +111,7 @@ def fine_tune_model(base_model_name, dataset_name):
         
     # Create trainer
 
-    trainer = Trainer(
+    trainer = Seq2SeqTrainer(
         model=peft_model,
         args=training_args,
         train_dataset=train_dataset,