Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import json
 from datetime import datetime
 from pathlib import Path
@@ -13,12 +14,18 @@ IMAGE_DATASET_DIR = Path("image_dataset") / f"train-{uuid4()}"
 IMAGE_DATASET_DIR.mkdir(parents=True, exist_ok=True)
 IMAGE_JSONL_PATH = IMAGE_DATASET_DIR / "metadata.jsonl"
 
+# Read Hugging Face token from secret (HF Spaces automatically sets it)
+HF_TOKEN = os.environ.get("HF_TOKEN")
+if not HF_TOKEN:
+    raise ValueError("HF_TOKEN not found! Please set it in HF Spaces Secrets.")
+
 # Scheduler for your HF dataset repo
 scheduler = CommitScheduler(
     repo_id="codewithRiz/Buck_data_saving",  # your dataset repo
     repo_type="dataset",
     folder_path=IMAGE_DATASET_DIR,
     path_in_repo=IMAGE_DATASET_DIR.name,
+    token=HF_TOKEN,  # pass the token here
 )
 
 # ---------------- Save uploaded image ----------------