Update Dockerfile
Browse files- Dockerfile +5 -5
Dockerfile
CHANGED
|
@@ -38,18 +38,18 @@ RUN poetry install --without dev --no-root
|
|
| 38 |
ENV PATH=$HOME/app/.venv/bin:$PATH
|
| 39 |
ENV PYTHONPATH=.
|
| 40 |
|
| 41 |
-
# download needed data from Hugging Face
|
| 42 |
-
RUN wget https://huggingface.co/datasets/lajota13/lfw_facenet_embeddings/resolve/main/lfw_season_embeddings_train.parquet -O data/lfw_season_embeddings_train.parquet
|
| 43 |
-
|
| 44 |
# download vggface2 weights for Facenet embeddings
|
| 45 |
RUN mkdir -p ~/.cache/torch/checkpoints
|
| 46 |
RUN wget https://github.com/timesler/facenet-pytorch/releases/download/v2.2.9/20180402-114759-vggface2.pt -O ~/.cache/torch/checkpoints/20180402-114759-vggface2.pt
|
| 47 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
# download season classifier weights
|
| 49 |
RUN wget https://huggingface.co/datasets/lajota13/lfw_facenet_embeddings/resolve/main/classifier_weights_v1.pt -O data/classifier_weights_v1.pt
|
| 50 |
|
| 51 |
-
COPY --chown=user . $HOME/app/
|
| 52 |
-
|
| 53 |
HEALTHCHECK CMD curl --fail http://localhost:$PORT/_stcore/health
|
| 54 |
|
| 55 |
SHELL ["/bin/bash", "-c"]
|
|
|
|
| 38 |
ENV PATH=$HOME/app/.venv/bin:$PATH
|
| 39 |
ENV PYTHONPATH=.
|
| 40 |
|
|
|
|
|
|
|
|
|
|
| 41 |
# download vggface2 weights for Facenet embeddings
|
| 42 |
RUN mkdir -p ~/.cache/torch/checkpoints
|
| 43 |
RUN wget https://github.com/timesler/facenet-pytorch/releases/download/v2.2.9/20180402-114759-vggface2.pt -O ~/.cache/torch/checkpoints/20180402-114759-vggface2.pt
|
| 44 |
|
| 45 |
+
COPY --chown=user . $HOME/app/
|
| 46 |
+
|
| 47 |
+
# download needed data from Hugging Face
|
| 48 |
+
RUN wget https://huggingface.co/datasets/lajota13/lfw_facenet_embeddings/resolve/main/lfw_season_embeddings_train.parquet -O data/lfw_season_embeddings_train.parquet
|
| 49 |
+
|
| 50 |
# download season classifier weights
|
| 51 |
RUN wget https://huggingface.co/datasets/lajota13/lfw_facenet_embeddings/resolve/main/classifier_weights_v1.pt -O data/classifier_weights_v1.pt
|
| 52 |
|
|
|
|
|
|
|
| 53 |
HEALTHCHECK CMD curl --fail http://localhost:$PORT/_stcore/health
|
| 54 |
|
| 55 |
SHELL ["/bin/bash", "-c"]
|