File size: 2,637 Bytes
3ccf12e 1af1777 e4a50e0 3ccf12e 1af1777 e4a50e0 3ccf12e 1af1777 3ccf12e 1af1777 3ccf12e 1af1777 3ccf12e 1af1777 3ccf12e 22ef630 208fa1c 3ccf12e 1af1777 3ccf12e 0855034 3ccf12e 22ef630 0855034 22ef630 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
# Use an official Python runtime as a parent image
FROM python:3.8
# Set the working directory in the container
WORKDIR $HOME/app
# Install Python dependencies
COPY ./requirements.txt /app/requirements.txt
RUN pip install --no-cache-dir -r requirements.txt
# Install system dependencies
RUN apt-get update && apt-get install -y \
build-essential \
cmake \
curl \
ca-certificates \
gcc \
git \
git-lfs \
wget \
libpq-dev \
libsndfile1-dev \
libgl1 \
unzip \
libjpeg-dev \
libpng-dev \
libgomp1 \
&& rm -rf /var/lib/apt/lists/* \
&& apt-get clean
WORKDIR $HOME/app
COPY --chown=user . $HOME/app
# Create .streamlit/config.toml file
RUN mkdir -p /app/.streamlit && \
echo "[server]\nenableXsrfProtection = false\nenableCORS = false" > /app/.streamlit/config.toml
# Clone the repository
RUN git clone https://github.com/nguyen-brat/text-remove.git $HOME/app/text-remove
# Set the working directory to the cloned repo
WORKDIR $HOME/app/text-remove
COPY --chown=user . $HOME/app/text-remove
RUN mkdir -p $HOME/app/text-remove/.streamlit && \
echo "[server]\nenableXsrfProtection = false\nenableCORS = false" > $HOME/app/text-remove/.streamlit/config.toml
# Set up Craft
WORKDIR $HOME/app/text-remove/CRAFT-pytorch
COPY --chown=user . $HOME/app/text-remove/CRAFT-pytorch
RUN wget --no-check-certificate 'https://docs.google.com/uc?export=download&id=1Jk4eGD7crsqCCg9C9VjCLkMN3ze8kutZ' -O "craft_mlt_25k.pth"
RUN pip install gdown
RUN gdown 1XSaFwBkOaFOdtk4Ane3DFyJGPRw6v5bO
# Set up lama
WORKDIR $HOME/app/text-remove/lama
COPY --chown=user . $HOME/app/text-remove/lama
RUN curl -LJO https://huggingface.co/smartywu/big-lama/resolve/main/big-lama.zip
RUN unzip big-lama.zip
# RUN useradd -m -u 1000 user
# USER user
# ENV HOME /home/user
# ENV PATH $HOME/.local/bin:$PATH
# Set the working directory back to the root of the project
WORKDIR $HOME/app/text-remove
RUN chmod 777 $HOME/app/text-remove/target_test
RUN chmod 777 $HOME/app/text-remove/test_folder
# Make port 7860 available to the world outside this container
EXPOSE 7860
# Run app.py when the container launches
ENTRYPOINT ["streamlit", "run"]
CMD ["app.py", "--server.port=7860", "--server.headless=true", "--server.enableCORS=false", "--server.enableXsrfProtection=false", "--server.fileWatcherType=none"]
# CMD ["streamlit", "run", "app.py", "--server.port=7860", "--server.address=0.0.0.0"]
#CMD streamlit run app.py \
# ---server.port=7860 \
# --server.headless true \
# --server.enableCORS false \
# --server.enableXsrfProtection false \
# --server.fileWatcherType none
|