bugfix
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
|
@@ -4,6 +4,7 @@ LABEL python_version=python3.10
|
|
| 4 |
RUN apt-get update && apt-get install -y \
|
| 5 |
build-essential \
|
| 6 |
curl \
|
|
|
|
| 7 |
software-properties-common \
|
| 8 |
git \
|
| 9 |
&& rm -rf /var/lib/apt/lists/*
|
|
@@ -37,6 +38,7 @@ COPY . .
|
|
| 37 |
RUN wget https://huggingface.co/datasets/lajota13/lfw_facenet_embeddings/resolve/main/lfw_season_embeddings_train.parquet -O data/lfw_season_embeddings_train.parquet
|
| 38 |
|
| 39 |
# download vggface2 weights for Facenet embeddings
|
|
|
|
| 40 |
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
|
| 41 |
|
| 42 |
HEALTHCHECK CMD curl --fail http://localhost:$PORT/_stcore/health
|
|
|
|
| 4 |
RUN apt-get update && apt-get install -y \
|
| 5 |
build-essential \
|
| 6 |
curl \
|
| 7 |
+
wget \
|
| 8 |
software-properties-common \
|
| 9 |
git \
|
| 10 |
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
| 38 |
RUN wget https://huggingface.co/datasets/lajota13/lfw_facenet_embeddings/resolve/main/lfw_season_embeddings_train.parquet -O data/lfw_season_embeddings_train.parquet
|
| 39 |
|
| 40 |
# download vggface2 weights for Facenet embeddings
|
| 41 |
+
RUN mkdir -p ~/.cache/torch/checkpoints
|
| 42 |
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
|
| 43 |
|
| 44 |
HEALTHCHECK CMD curl --fail http://localhost:$PORT/_stcore/health
|