jclyo1's picture
gpu work
8ba460b
raw
history blame
1.14 kB
FROM python:3.9
# Using the Ubuntu image (our OS)
# Update package manager (apt-get)
# and install (with the yes flag `-y`)
# Python and Pip
RUN apt-get update && apt-get install -y \
python3-pip \
wget
WORKDIR /code
COPY ./requirements.txt /code/requirements.txt
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
RUN pip install diffusers transformers accelerate scipy safetensors
# Set up a new user named "user" with user ID 1000
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
# Download and uzip truepic-size from Google Drive
RUN pip install gdown
RUN gdown --id 1MdXMu8xSuG8WClY3JSgtL3Vy1IWTmWZO
RUN tar -xf truepic-sign-v0.1.0-ubuntu-latest.tar.gz
RUN chmod +x truepic-sign
WORKDIR $HOME/app
# COPY . .
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]