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, pipeline
+from transformers import AutoModelForCausalLM, AutoTokenizer, Seq2SeqTrainer, Seq2SeqTrainingArguments, Trainer, pipeline
 
 ACTION_1 = "Prompt base model"
 ACTION_2 = "Fine-tune base model"
@@ -103,16 +103,16 @@ def fine_tune_model(base_model_name, dataset_name):
         task_type=TaskType.SEQ_2_SEQ_LM,
     )
     
-    model = get_peft_model(model, lora_config)
+    peft_model = get_peft_model(model, lora_config)
     
     print("### PEFT")
-    model.print_trainable_parameters()
+    peft_model.print_trainable_parameters()
     print("###")
         
     # Create trainer
 
-    trainer = Seq2SeqTrainer(
-        model=model,
+    trainer = Trainer(
+        model=peft_model,
         args=training_args,
         train_dataset=train_dataset,
         eval_dataset=eval_dataset,