Dockerfile

FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
# FROM python:3.10.9

# Use a more secure method for setting the working directory
WORKDIR /app

# Copy the requirements file
COPY ./requirements.txt /app/requirements.txt

# Install dependencies
RUN apt update && apt install -y python3 python3-pip

RUN python3 -m pip install --no-cache-dir --upgrade -r /app/requirements.txt
RUN apt update && apt install -y ffmpeg
# Add and configure the user
RUN useradd -m -u 1000 user

# Set environment variables for the user
ENV HOME=/home/user \
 PATH=/home/user/.local/bin:$PATH

# Set the user and working directory
USER user
WORKDIR $HOME/app

# Copy the application code
COPY --chown=user . $HOME/app

# Ensure the user has the correct permissions
RUN chmod -R 777 $HOME/app

# Set additional environment variables
ENV NUMBA_CACHE_DIR=/tmp/NUMBA_CACHE_DIR/
ENV MPLCONFIGDIR=/tmp/MPLCONFIGDIR/
ENV HOME=/home/user
ENV PATH="/opt/venv/bin:$PATH"

# Set the command to run the application
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]