build(docker): on CI Docker Job, build for Stage prod_install (--targ…

…et), since missing vgg weights

.github/workflows/test.yaml

@@ -512,8 +512,13 @@ jobs:
       run: |
         # workflow enabled for branches and v* tags
         IMAGE_TAG="${GITHUB_REF_NAME}"  # this is branch name or tag name
-        echo "IMAGE_REF=neural-style-transfer:${IMAGE_TAG}" >> $GITHUB_ENV
-        docker build -t "${DOCKER_USER}/neural-style-transfer:${IMAGE_TAG}" .
+
+        echo "IMAGE_REF=neural-style-transfer-cli:${IMAGE_TAG}" >> $GITHUB_ENV
+
+        # build an image that does not have vgg model weights
+        # TODO: add image model weights file in this env somehow
+        
+        docker build --target prod_install -t "${DOCKER_USER}/neural-style-transfer:${IMAGE_TAG}" .
     - name: Publish Docker Image to DockerHub
       env:
         DOCKER_PASSWORD: ${{ secrets.DOCKER_PASSWORD }}

Dockerfile

@@ -51,12 +51,19 @@ RUN pip install --no-cache-dir --user ./dist/*.whl
 
 # Optionaly, add the CLI executable to the PATH
 # to make the `nst` CLI available in the image
-# ENV PATH="/root/.local/bin:$PATH"
+ENV PATH="/root/.local/bin:$PATH"
 
 ## This is a Builder kind of Stage
 ## Consider using the docker feature on-build
 
-### END of Prod BUILD ###
+# build: docker build --target prod_install -t nst_cli .
+# Usable as: docker run -it --rm nst_cli --entrypoint nst
+
+# EG: docker run -it --rm nst_cli nst --help
+
+CMD [ "nst" ]
+
+### END of Prod Build Installation ###