Dockerfile

FROM python:3.10-slim

RUN apt-get update && apt-get install -y \
    git ffmpeg libsm6 libxext6 libgl1 libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*

RUN pip install --no-cache-dir \
    gradio==4.44.0 \
    torch==2.2.0 \
    torchvision==0.17.0 \
    transformers==4.40.0 \
    huggingface_hub==0.23.0 \
    facenet-pytorch==2.5.3 \
    Pillow==10.3.0 \
    opencv-python-headless==4.9.0.80 \
    numpy==1.26.4 \
    matplotlib==3.8.4

RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user PATH=/home/user/.local/bin:$PATH
WORKDIR /home/user/app
COPY --chown=user . .

EXPOSE 7860
CMD ["python", "app.py"]