workdir COPY --chown

Dockerfile

@@ -19,9 +19,11 @@ RUN echo "source /opt/venv/bin/activate" >> /root/.bashrc
 
 # 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 "." \
     --extra-index-url https://download.pytorch.org/whl/cpu
 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_lego_simple_v2", "--port 9200"]