diff --git a/lerobot/html_dataset_visualizer/src/app/[org]/[dataset]/[episode]/fetch-data.ts b/lerobot/html_dataset_visualizer/src/app/[org]/[dataset]/[episode]/fetch-data.ts
index e5056b985..54a098061 100644
--- a/lerobot/html_dataset_visualizer/src/app/[org]/[dataset]/[episode]/fetch-data.ts
+++ b/lerobot/html_dataset_visualizer/src/app/[org]/[dataset]/[episode]/fetch-data.ts
@@ -10,6 +10,8 @@ import { pick } from "@/utils/pick";
const DATASET_URL =
process.env.DATASET_URL || "https://huggingface.co/datasets";
+const DEFAULT_REVISION = "v2.1";
+
const SERIES_NAME_DELIMITER = " | ";
export async function getEpisodeData(
@@ -20,7 +22,7 @@ export async function getEpisodeData(
const repoId = `${org}/${dataset}`;
try {
const episode_chunk = Math.floor(0 / 1000);
- const jsonUrl = `${DATASET_URL}/${repoId}/resolve/main/meta/info.json`;
+ const jsonUrl = `${DATASET_URL}/${repoId}/resolve/${DEFAULT_REVISION}/meta/info.json`;
const info = await fetchJson(jsonUrl);
@@ -56,7 +58,7 @@ export async function getEpisodeData(
});
return {
filename: key,
- url: `${DATASET_URL}/${repoId}/resolve/main/` + videoPath,
+ url: `${DATASET_URL}/${repoId}/resolve/${DEFAULT_REVISION}/` + videoPath,
};
});
@@ -103,7 +105,7 @@ export async function getEpisodeData(
});
const parquetUrl =
- `${DATASET_URL}/${repoId}/resolve/main/` +
+ `${DATASET_URL}/${repoId}/resolve/${DEFAULT_REVISION}/` +
formatStringWithVars(info.data_path, {
episode_chunk: episode_chunk.toString().padStart(3, "0"),
episode_index: episodeId.toString().padStart(6, "0"),
diff --git a/lerobot/html_dataset_visualizer/src/app/explore/page.tsx b/lerobot/html_dataset_visualizer/src/app/explore/page.tsx
index 5a175f028..d3ad52f69 100644
--- a/lerobot/html_dataset_visualizer/src/app/explore/page.tsx
+++ b/lerobot/html_dataset_visualizer/src/app/explore/page.tsx
@@ -38,6 +38,9 @@ export default async function ExplorePage({
return Failed to load datasets.
;
}
+ // Default to v2.1 revision for dataset loading
+ const DEFAULT_REVISION = "v2.1";
+
// Fetch episode 0 data for each dataset
const datasetWithVideos = (
await Promise.all(
@@ -45,7 +48,7 @@ export default async function ExplorePage({
try {
const [org, dataset] = ds.id.split("/");
const repoId = `${org}/${dataset}`;
- const jsonUrl = `https://huggingface.co/datasets/${repoId}/resolve/main/meta/info.json`;
+ const jsonUrl = `https://huggingface.co/datasets/${repoId}/resolve/${DEFAULT_REVISION}/meta/info.json`;
const info = await fetchJson(jsonUrl);
const videoEntry = Object.entries(info.features).find(
([key, value]) => value.dtype === "video",
@@ -59,7 +62,7 @@ export default async function ExplorePage({
episode_index: "0".padStart(6, "0"),
});
const url =
- `https://huggingface.co/datasets/${repoId}/resolve/main/` +
+ `https://huggingface.co/datasets/${repoId}/resolve/${DEFAULT_REVISION}/` +
videoPath;
// Check if videoUrl exists (status 200)
try {