Tensor-Pruning / Dockerfile
Angshul's picture
Upload 14 files
a42d9ad verified
Raw History Blame Contribute Delete
578 Bytes
FROM python:3.12-slim-bookworm
ENV PYTHONUNBUFFERED=1 PIP_NO_CACHE_DIR=1 PYTHONDONTWRITEBYTECODE=1
RUN apt-get update \
&& apt-get install -y --no-install-recommends g++ \
&& rm -rf /var/lib/apt/lists/* \
&& useradd -m -u 1000 user \
&& mkdir -p /models /home/user/work \
&& chown -R user:user /home/user/work
WORKDIR /app
COPY pyproject.toml README.md space_app.py ./
COPY tensor_pruning/ ./tensor_pruning/
RUN python -m pip install '.[space]'
USER user
EXPOSE 7860
CMD ["python", "-m", "uvicorn", "space_app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]