update

Dockerfile

@@ -2,7 +2,11 @@
 FROM python:3.8
 
 # Set the working directory in the container
-WORKDIR /app
+WORKDIR $HOME/app
+COPY --chown=user . $HOME/app
+
+RUN useradd -m -u 1000 user
+USER user
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -24,40 +28,42 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/* \
     && apt-get clean
 
-WORKDIR /app
-COPY --chown=user . /app
+WORKDIR $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 /app/text-remove
+RUN git clone https://github.com/nguyen-brat/text-remove.git $HOME/app/text-remove
 
 # Set the working directory to the cloned repo
-WORKDIR /app/text-remove
-COPY --chown=user . /app/text-remove
-RUN mkdir -p /app/text-remove/.streamlit && \
-    echo "[server]\nenableXsrfProtection = false\nenableCORS = false" > /app/text-remove/.streamlit/config.toml
+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
 
 # Install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Set up Craft
-WORKDIR /app/text-remove/CRAFT-pytorch
-COPY --chown=user . /app/text-remove/CRAFT-pytorch
+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 /app/text-remove/lama
-COPY --chown=user . /app/text-remove/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
 
 # Set the working directory back to the root of the project
-WORKDIR /app/text-remove
+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