cadene HF staff commited on
Commit
76c5e2d
1 Parent(s): 37e79a4

Add cadene/visualize_dataset_koch_bimanual_folding

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -20,7 +20,7 @@ RUN echo "source /opt/venv/bin/activate" >> /root/.bashrc
20
  RUN useradd -m -u 1000 user
21
 
22
  # Install LeRobot
23
- RUN git clone --branch user/rcadene/2024_05_17_improve_visualize_dataset https://github.com/huggingface/lerobot.git /lerobot
24
  WORKDIR /lerobot
25
  RUN pip install --upgrade --no-cache-dir pip
26
  RUN pip install --no-cache-dir "." \
@@ -28,4 +28,4 @@ RUN pip install --no-cache-dir "." \
28
  RUN pip install --no-cache-dir flask
29
 
30
  COPY --chown=user . /lerobot
31
- 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"]
 
20
  RUN useradd -m -u 1000 user
21
 
22
  # Install LeRobot
23
+ RUN git clone --branch main https://github.com/huggingface/lerobot.git /lerobot
24
  WORKDIR /lerobot
25
  RUN pip install --upgrade --no-cache-dir pip
26
  RUN pip install --no-cache-dir "." \
 
28
  RUN pip install --no-cache-dir flask
29
 
30
  COPY --chown=user . /lerobot
31
+ CMD ["python", "lerobot/scripts/visualize_dataset_html.py", "--repo-id", "cadene/visualize_dataset_koch_bimanual_folding", "--host", "0.0.0.0", "--port", "7860", "--output-dir", "/tmp/outputs/visualize_dataset_html/cadene/visualize_dataset_koch_bimanual_folding"]