lajota13 commited on
Commit
e748f03
·
1 Parent(s): 49013e7
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -48,6 +48,9 @@ RUN wget https://huggingface.co/datasets/lajota13/lfw_facenet_embeddings/resolve
48
  RUN mkdir -p ~/.cache/torch/checkpoints
49
  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
50
 
 
 
 
51
  HEALTHCHECK CMD curl --fail http://localhost:$PORT/_stcore/health
52
 
53
  SHELL ["/bin/bash", "-c"]
 
48
  RUN mkdir -p ~/.cache/torch/checkpoints
49
  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
50
 
51
+ # download season classifier weights
52
+ wget https://huggingface.co/datasets/lajota13/lfw_facenet_embeddings/resolve/main/classifier_weights_v1.pt -O data/classifier_weights_v1.pt
53
+
54
  HEALTHCHECK CMD curl --fail http://localhost:$PORT/_stcore/health
55
 
56
  SHELL ["/bin/bash", "-c"]