import os
from pathlib import Path
from huggingface_hub import snapshot_download

data_dir = Path("/tmp/trackio_data")
data_dir.mkdir(exist_ok=True)

snapshot_download(
    repo_id="kashif/xtoken-distillation-bucket",
    repo_type="dataset",
    local_dir=str(data_dir),
    ignore_patterns=[".gitattributes", "README.md"],
)

os.environ["TRACKIO_DIR"] = str(data_dir)
import trackio
trackio.show(project="xtoken-distillation", host="0.0.0.0")