From 9e9c1dfff6664b56d3e57243df850ad63c5fe608 Mon Sep 17 00:00:00 2001 From: silvandeleemput Date: Thu, 23 Nov 2023 21:30:45 +0100 Subject: [PATCH] add mhub model definition import to Dockerfile --- models/gc_node21_baseline/dockerfiles/Dockerfile | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/models/gc_node21_baseline/dockerfiles/Dockerfile b/models/gc_node21_baseline/dockerfiles/Dockerfile index 4a756a55..304be8a4 100644 --- a/models/gc_node21_baseline/dockerfiles/Dockerfile +++ b/models/gc_node21_baseline/dockerfiles/Dockerfile @@ -27,12 +27,9 @@ RUN git clone https://github.com/node21challenge/node21_detection_baseline.git / # Install Node21 additional dependencies RUN pip3 install --no-cache-dir evalutils==0.2.4 -# TODO: FIXME: temporary fix waiting for the latest base image update -#RUN git stash \ -# && git fetch https://github.com/MHubAI/models.git main \ -# && git merge FETCH_HEAD \ -# && git sparse-checkout set "models/gc_node21_baseline" \ -# && git fetch https://github.com/MHubAI/models.git main +# Import the MHub model definiton +ARG MHUB_MODELS_REPO +RUN buildutils/import_mhub_model.sh gc_node21_baseline ${MHUB_MODELS_REPO} # Configure app /opt/algorithm to be the torch hub directory to pick up the model weights correctly ENV TORCH_HOME /opt/algorithm