FROM python:3.11-slim | |
### Set up user with permissions | |
# Set up a new user named "user" with user ID 1000 | |
RUN apt-get update && apt-get install -y python3-opencv | |
RUN pip install opencv-python | |
RUN useradd -m -u 1000 user | |
# Switch to the "user" user | |
USER user | |
# Set home to the user's home directory | |
ENV HOME=/home/user \ | |
PATH=/home/user/.local/bin:$PATH | |
# Set the working directory to the user's home directory | |
WORKDIR $HOME/app | |
# Copy the current directory contents into the container at $HOME/app setting the owner to the user | |
COPY --chown=user . $HOME/app | |
### Set up app-specific content | |
COPY req.txt req.txt | |
# Install basicsr - https://github.com/xinntao/BasicSR | |
# We use BasicSR for both training and inference | |
RUN pip install basicsr | |
# Install facexlib - https://github.com/xinntao/facexlib | |
# We use face detection and face restoration helper in the facexlib package | |
RUN pip install facexlib | |
RUN pip3 install -r req.txt | |
RUN python setup.py develop | |
# If you want to enhance the background (non-face) regions with Real-ESRGAN, | |
# you also need to install the realesrgan package | |
RUN pip install realesrgan | |
COPY . . | |
### Update permissions for the app | |
USER root | |
RUN chmod 777 ~/app/* | |
USER user | |
EXPOSE 7860 7860 | |
ENTRYPOINT ["streamlit", "run"] | |
CMD ["streamlit-app.py", "--server.port", "7860"] |