Spaces:

lerobot
/

visualize_dataset

Running

mishig HF staff commited on Sep 3, 2024

Commit

a7a6937

verified ·

1 Parent(s): 37e79a4

Visualize all datasets

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -20,7 +20,7 @@ RUN echo "source /opt/venv/bin/activate" >> /root/.bashrc
 RUN useradd -m -u 1000 user
 # Install LeRobot
-RUN git clone --branch user/rcadene/2024_05_17_improve_visualize_dataset https://github.com/huggingface/lerobot.git /lerobot
 WORKDIR /lerobot
 RUN pip install --upgrade --no-cache-dir pip
 RUN pip install --no-cache-dir "." \
@@ -28,4 +28,4 @@ RUN pip install --no-cache-dir "." \
 RUN pip install --no-cache-dir flask
 COPY --chown=user . /lerobot
-CMD ["python", "lerobot/scripts/visualize_dataset_html.py", "--repo-id", "lerobot/koch_pick_place_1_lego", "--host", "0.0.0.0", "--port", "7860", "--output-dir", "/tmp/outputs/visualize_dataset_html/lerobot/koch_pick_place_1_lego"]

 RUN useradd -m -u 1000 user
 # Install LeRobot
+RUN git clone --branch visualize_all_datastes https://github.com/mishig25/lerobot.git /lerobot
 WORKDIR /lerobot
 RUN pip install --upgrade --no-cache-dir pip
 RUN pip install --no-cache-dir "." \
 RUN pip install --no-cache-dir flask
 COPY --chown=user . /lerobot
+CMD ["python", "lerobot/scripts/visualize_dataset_html.py", "--host", "0.0.0.0", "--port", "7860"]