ImageColoriser / Dockerfile
Divya-A's picture
ImageColoriser: Flask app and colorization models for Hugging Face Space
e6e4be7
raw
history blame contribute delete
742 Bytes
# Hugging Face Docker Space: https://huggingface.co/docs/hub/spaces-sdks-docker
FROM python:3.10-slim
RUN apt-get update && apt-get install -y --no-install-recommends \
libgl1 \
libglib2.0-0 \
&& rm -rf /var/lib/apt/lists/*
RUN useradd -m -u 1000 user
USER user
ENV PATH="/home/user/.local/bin:$PATH"
WORKDIR /app
COPY --chown=user requirements.txt requirements.txt
RUN pip install --no-cache-dir --upgrade pip \
&& pip install --no-cache-dir -r requirements.txt \
--index-url https://download.pytorch.org/whl/cpu \
--extra-index-url https://pypi.org/simple
COPY --chown=user . /app
EXPOSE 7860
CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "--threads", "2", "--timeout", "180", "app:app"]