update

Dockerfile

@@ -20,11 +20,11 @@ RUN echo "source /opt/venv/bin/activate" >> /root/.bashrc
 # Install LeRobot
 RUN git clone https://github.com/huggingface/lerobot.git /lerobot
 WORKDIR /lerobot
+RUN git fetch origin user/rcadene/2024_05_17_improve_visualize_dataset
+RUN git checkout origin/user/rcadene/2024_05_17_improve_visualize_dataset
 RUN pip install --upgrade --no-cache-dir pip
 RUN pip install --no-cache-dir "." \
     --extra-index-url https://download.pytorch.org/whl/cpu
 RUN pip install --no-cache-dir flask
 
-RUN ls .
-RUN pwd
 RUN python lerobot/scripts/visualize_dataset_html.py --repo-id lerobot/koch_pick_place_lego_simple_v2 --port 9200