Update app.py

app.py

@@ -104,7 +104,10 @@ def fine_tune_model(base_model_name, dataset_name):
     )
     
     model = get_peft_model(model, lora_config)
+    
+    print("### PEFT")
     model.print_trainable_parameters()
+    print("###")
         
     # Create trainer