Update train.py

train.py

@@ -5,6 +5,8 @@ from datasets import Dataset
 from transformers import TrainingArguments
 from unsloth import FastLanguageModel
 from trl import SFTTrainer  # ✅ now works because we added 'trl'
+from shutil import copytree
+import os
 
 
 # Load and format your dataset
@@ -57,4 +59,8 @@ trainer = SFTTrainer(
 trainer.train()
 
 # Save the fine-tuned LoRA adapter
-model.save_pretrained("./lora-finetuned")
+model.save_pretrained("./lora-finetuned")
+
+# Copy LoRA output directory to current visible path
+if os.path.exists("output") and os.path.isdir("output"):
+    copytree("output", "/home/user/app/final_model", dirs_exist_ok=True)