nguyen-brat commited on
Commit
1683dda
·
1 Parent(s): 03af360
Files changed (1) hide show
  1. Dockerfile +4 -2
Dockerfile CHANGED
@@ -25,7 +25,7 @@ RUN apt-get update && apt-get install -y \
25
  && apt-get clean
26
 
27
  WORKDIR /app
28
-
29
  # Create .streamlit/config.toml file
30
  RUN mkdir -p /app/.streamlit && \
31
  echo "[server]\nenableXsrfProtection = false\nenableCORS = false" > /app/.streamlit/config.toml
@@ -35,7 +35,7 @@ RUN git clone https://github.com/nguyen-brat/text-remove.git /app/text-remove
35
 
36
  # Set the working directory to the cloned repo
37
  WORKDIR /app/text-remove
38
-
39
  RUN mkdir -p /app/text-remove/.streamlit && \
40
  echo "[server]\nenableXsrfProtection = false\nenableCORS = false" > /app/text-remove/.streamlit/config.toml
41
 
@@ -45,12 +45,14 @@ RUN pip install --no-cache-dir -r requirements.txt
45
 
46
  # Set up Craft
47
  WORKDIR /app/text-remove/CRAFT-pytorch
 
48
  RUN wget --no-check-certificate 'https://docs.google.com/uc?export=download&id=1Jk4eGD7crsqCCg9C9VjCLkMN3ze8kutZ' -O "craft_mlt_25k.pth"
49
  RUN pip install gdown
50
  RUN gdown 1XSaFwBkOaFOdtk4Ane3DFyJGPRw6v5bO
51
 
52
  # Set up lama
53
  WORKDIR /app/text-remove/lama
 
54
  RUN curl -LJO https://huggingface.co/smartywu/big-lama/resolve/main/big-lama.zip
55
  RUN unzip big-lama.zip
56
 
 
25
  && apt-get clean
26
 
27
  WORKDIR /app
28
+ COPY --chown=user . /app
29
  # Create .streamlit/config.toml file
30
  RUN mkdir -p /app/.streamlit && \
31
  echo "[server]\nenableXsrfProtection = false\nenableCORS = false" > /app/.streamlit/config.toml
 
35
 
36
  # Set the working directory to the cloned repo
37
  WORKDIR /app/text-remove
38
+ COPY --chown=user . /app/text-remove
39
  RUN mkdir -p /app/text-remove/.streamlit && \
40
  echo "[server]\nenableXsrfProtection = false\nenableCORS = false" > /app/text-remove/.streamlit/config.toml
41
 
 
45
 
46
  # Set up Craft
47
  WORKDIR /app/text-remove/CRAFT-pytorch
48
+ COPY --chown=user . /app/text-remove/CRAFT-pytorch
49
  RUN wget --no-check-certificate 'https://docs.google.com/uc?export=download&id=1Jk4eGD7crsqCCg9C9VjCLkMN3ze8kutZ' -O "craft_mlt_25k.pth"
50
  RUN pip install gdown
51
  RUN gdown 1XSaFwBkOaFOdtk4Ane3DFyJGPRw6v5bO
52
 
53
  # Set up lama
54
  WORKDIR /app/text-remove/lama
55
+ COPY --chown=user . /app/text-remove/lama
56
  RUN curl -LJO https://huggingface.co/smartywu/big-lama/resolve/main/big-lama.zip
57
  RUN unzip big-lama.zip
58