CADGenBench / Dockerfile
Michael Rabinovich
Dockerfile: bump CADGENBENCH_SHA to 9bd02fb (lazy baseline import)
67fd44c
Raw
History Blame
2.35 kB
# syntax=docker/dockerfile:1.7
#
# Hugging Face Space (sdk: docker) for the CADGenBench leaderboard.
#
# HF builds this server-side on each git push. Local smoke test:
#
# docker buildx build --platform linux/amd64 -t cadgenbench-space-test .
#
# cadgenbench is installed from github.com/huggingface/cadgenbench, which
# is Public. No build secrets or auth required.
FROM python:3.12-slim-bookworm
ENV PYTHONUNBUFFERED=1 \
PYTHONDONTWRITEBYTECODE=1 \
PIP_DISABLE_PIP_VERSION_CHECK=1 \
PLAYWRIGHT_BROWSERS_PATH=/opt/ms-playwright \
GRADIO_SERVER_NAME=0.0.0.0 \
GRADIO_SERVER_PORT=7860
# OS deps:
# git, ca-certificates -> pip install from git+https://
# libgl1, libglib2.0-0, libsm6,
# libxext6, libxrender1, -> OCP / build123d / Pillow runtime
# libgomp1, libfontconfig1
# Chromium's own runtime libs are added by `playwright install --with-deps`
# below, so they aren't repeated here.
RUN apt-get update && apt-get install -y --no-install-recommends \
git ca-certificates \
libgl1 libglib2.0-0 libsm6 libxext6 libxrender1 libgomp1 libfontconfig1 \
&& rm -rf /var/lib/apt/lists/*
# Space-side Python deps (gradio, pandas, huggingface_hub, datasets).
COPY requirements.txt /tmp/requirements.txt
RUN pip install --no-cache-dir -r /tmp/requirements.txt \
&& rm /tmp/requirements.txt
# Playwright + headless Chromium for cadgenbench's STEP -> PNG renderer
# (cadgenbench.common.viewer wraps tcv-screenshots -> playwright). Browsers
# land in /opt/ms-playwright (world-readable) so the non-root runtime user
# below can find them via PLAYWRIGHT_BROWSERS_PATH. ~200 MB layer.
RUN pip install --no-cache-dir playwright \
&& playwright install --with-deps chromium
# cadgenbench from the Public GitHub repo, pinned to a commit. Bumping
# CADGENBENCH_SHA is the one-line path to picking up a new cadgenbench.
ARG CADGENBENCH_SHA=9bd02fb
RUN pip install --no-cache-dir \
"cadgenbench @ git+https://github.com/huggingface/cadgenbench.git@${CADGENBENCH_SHA}"
# Drop privileges. HF Spaces conventionally run as uid 1000 with
# WORKDIR /home/user/app.
RUN useradd -m -u 1000 user \
&& mkdir -p /home/user/app \
&& chown -R user:user /home/user/app
USER user
WORKDIR /home/user/app
COPY --chown=user:user *.py results.jsonl ./
EXPOSE 7860
CMD ["python", "app.py"]