Spaces:

cadene
/

visualize_dataset_koch_bimanual_folding

Running

cadene HF staff commited on Jul 17

Commit

89090d4

•

1 Parent(s): 932b9b2

update

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -18,13 +18,11 @@ ENV PATH="/opt/venv/bin:$PATH"
 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 python lerobot/scripts/visualize_dataset_html.py --repo-id lerobot/koch_pick_place_lego_simple_v2 --port 9200

 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
+CMD ["python", "lerobot/scripts/visualize_dataset_html.py", "--repo-id", "lerobot/koch_pick_place_lego_simple_v2", "--port 9200"]