Update train.py

train.py

@@ -1,6 +1,13 @@
 from datasets import load_dataset
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, Trainer, TrainingArguments
 import evaluate
+from huggingface_hub import login
+
+# 🔑 Login to Hugging Face (add HF_TOKEN as a secret in your Space settings)
+import os
+hf_token = os.getenv("HF_TOKEN")
+if hf_token:
+    login(hf_token)
 
 # 1. Load BiScope dataset
 dataset = load_dataset("HanxiGuo/BiScope_Data")
@@ -14,7 +21,7 @@ def preprocess(examples):
 
 encoded_dataset = dataset.map(preprocess, batched=True)
 
-# 3. Load model
+# 3. Model
 model = AutoModelForSequenceClassification.from_pretrained(MODEL, num_labels=2)
 
 # 4. Metrics
@@ -35,8 +42,8 @@ training_args = TrainingArguments(
     per_device_eval_batch_size=16,
     num_train_epochs=2,
     weight_decay=0.01,
-    push_to_hub=True,  # ✅ Upload to HF
-    hub_model_id="yagnik12/AI_Text_Detecter_HanxiGuo_BiScope-Data"
+    push_to_hub=True,
+    hub_model_id="yagnik12/biscope-detector"
 )
 
 # 6. Trainer