Update train.py

train.py

@@ -57,28 +57,15 @@ trainer = SFTTrainer(
 
 trainer.train()
 
-# Save the fine-tuned LoRA adapter
-model.save_pretrained("./lora-finetuned")
+# ✅ Save to Hugging Face persistent path
+output_path = "./outputs/final_model"
+os.makedirs(output_path, exist_ok=True)
 
-# Copy to final_model AND Hugging Face root directory
-src_dir = "./lora-finetuned"
-dst_dir = "/home/user/app/final_model"
-root_dir = "/home/user/app"
-os.makedirs(dst_dir, exist_ok=True)
-
-if os.path.exists(src_dir):
-    for fname in os.listdir(src_dir):
-        fpath = os.path.join(src_dir, fname)
-        if os.path.isfile(fpath):
-            shutil.copy2(fpath, dst_dir)
-            print(f"✅ Copied {fname} to final_model")
-
-            try:
-                shutil.copy2(fpath, root_dir)
-                print(f"✅ Copied {fname} to app root for UI visibility")
-            except Exception as e:
-                print(f"⚠️ Failed to copy {fname} to root: {e}")
-        else:
-            print(f"📂 Skipped {fname} (not a file)")
-else:
-    print("❌ Source folder does not exist")
+try:
+    model.save_pretrained(output_path)
+    print(f"✅ Model saved to {output_path}")
+    print("📁 Contents of saved model directory:")
+    for file in os.listdir(output_path):
+        print(" -", file)
+except Exception as e:
+    print(f"❌ Failed to save model: {e}")