Update train.py

train.py

@@ -1,3 +1,20 @@
+# ==========================================
+# 0. SIMPLE AUTHENTICATION (FOR GATED MODEL)
+# ==========================================
+import os
+import sys
+from huggingface_hub import login
+
+# Grab the secret you set in the HF Space UI
+hf_token = os.environ.get("HF_TOKEN")
+
+if not hf_token:
+    print("❌ ERROR: HF_TOKEN not found in Secrets. Llama-3.2 download will fail.")
+    sys.exit(1)
+
+# Silently log in so Unsloth can download the gated weights
+login(token=hf_token.strip())
+print("✅ Authenticated. Ready to load Llama-3.2.")
 
 # ==========================================
 # 1. THE STABLE ENVIRONMENT INSTALL