Update app.py

app.py

@@ -106,13 +106,13 @@ def fine_tune_model(base_model_name, dataset_name):
     peft_model = get_peft_model(model, lora_config)
     
     print("### PEFT")
-    peft_model.print_trainable_parameters()
+    peft_model.print_trainable_parameters() # trainable params: 6,815,744 || all params: 8,037,076,992 || trainable%: 0.0848
     print("###")
         
     # Create trainer
 
     trainer = Seq2SeqTrainer(
-        model=peft_model,
+        model=model,
         args=training_args,
         train_dataset=train_dataset,
         eval_dataset=eval_dataset,