refactor backend dockerfile

cainky · Sep 23, 2023 · 4e318bf · 4e318bf
1 parent e13f4bb
commit 4e318bf
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/backend/docker/Dockerfile.backend b/backend/docker/Dockerfile.backend
@@ -22,16 +22,20 @@ RUN pip install gdown && \
     pip install -r requirements.txt
 
 # Download face detection model
-RUN mkdir -p face_detection/detection/sfd && \
-    wget "https://www.adrianbulat.com/downloads/python-fan/s3fd-619a316812.pth" -O "face_detection/detection/sfd/s3fd.pth"
+RUN if [ ! -f "face_detection/detection/sfd/s3fd.pth" ]; then \
+        mkdir -p face_detection/detection/sfd && \
+        wget "https://www.adrianbulat.com/downloads/python-fan/s3fd-619a316812.pth" -O "face_detection/detection/sfd/s3fd.pth"; \
+    fi
 
 # Create directory for checkpoints
 RUN mkdir -p checkpoints
 
-# Copy the download models script and run it
-COPY backend/docker/download_models.py .
-RUN chmod +x download_models.py && \
-    python download_models.py
+RUN if [ ! -f "checkpoints/wav2lip_gan.pth" ] || [ ! -f "checkpoints/face_segmentation.pth" ] || [ ! -f "checkpoints/esrgan_max.pth" ]; then \
+        # Copy the download models script and run it
+        COPY backend/docker/download_models.py . \
+        RUN chmod +x download_models.py && \
+        python download_models.py; \
+    fi
 
 # Create directories for input and output videos
 RUN mkdir -p videos results

diff --git a/backend/wav2lip-hq/.gitignore b/backend/wav2lip-hq/.gitignore
@@ -1,2 +1,3 @@
 checkpoints/*.pth
-temp/*
+temp/*.avi
+face_detection/detection/sfd/*.pth