Mini-Transformer / Dockerfile.infer
AlaBoussoffara's picture
added infer docker & /config option & various bug fixes & new tests
b22e8ba
Raw
History Blame Contribute Delete
531 Bytes
# syntax=docker/dockerfile:1
FROM python:3.10-slim
ENV DEBIAN_FRONTEND=noninteractive \
PIP_NO_CACHE_DIR=1 \
PYTHONDONTWRITEBYTECODE=1 \
PYTHONUNBUFFERED=1
RUN apt-get update && apt-get install -y --no-install-recommends \
git build-essential && \
rm -rf /var/lib/apt/lists/*
WORKDIR /app
COPY pyproject.toml README.md LICENSE ./
COPY src ./src
RUN pip install -U pip && pip install -e .[server,viz,notebook]
RUN mkdir -p trained_models
CMD ["mini-transformer-ui", "--host", "0.0.0.0", "--port", "8000"]