Update train.py

train.py

@@ -6,6 +6,16 @@ from datasets import Dataset
 import os
 from huggingface_hub import login
 
+# Get token from environment variables (set via Secrets)
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+# Authenticate with Hugging Face
+if HF_TOKEN:
+    login(token=HF_TOKEN)
+else:
+    raise ValueError("Hugging Face token not found. Make sure you added it in 'Secrets'.")
+
+
 
 
 # ✅ Load Extracted Data