Update app.py

app.py

@@ -249,18 +249,23 @@ def set_intro(css):
     
 from datetime import datetime
 from pathlib import Path
-from huggingface_hub import CommitScheduler
+from huggingface_hub import HfApi, CommitScheduler
 from uuid import uuid4
 
-JSON_DATASET_DIR = Path("json_dataset")
+# Access token from environment variable
+api = HfApi()
+api.set_access_token(os.getenv("HF_TOKEN"))
+
+JSON_DATASET_DIR = Path("json_demo_selected_io")
 JSON_DATASET_DIR.mkdir(parents=True, exist_ok=True)
 JSON_DATASET_PATH = JSON_DATASET_DIR / f"train-{uuid4()}.json"
 
 scheduler = CommitScheduler(
-    repo_id="example-space-to-dataset-json",
+    repo_id="demo-retriever",
     repo_type="dataset",
     folder_path=JSON_DATASET_DIR,
     path_in_repo="data",
+    hf_api=api
 )