label images

- part of #81
premAI-io · Aug 24, 2023 · 749f502 · 749f502
1 parent 719bf08
commit 749f502
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/utils.sh b/utils.sh
@@ -11,6 +11,7 @@ build_cpu(){
   docker buildx build ${@:3} \
     --file ./docker/cpu/Dockerfile \
     --build-arg="MODEL_ID=$MODEL_ID" \
+    --label="org.opencontainers.image.source=https://github.com/premAI-io/prem-services"
     --tag $IMAGE:latest \
     --tag $IMAGE:$VERSION \
     --platform ${BUILDX_PLATFORM:-linux/arm64,linux/amd64} \
@@ -23,6 +24,7 @@ build_gpu(){
   docker buildx build ${@:3} \
     --file ./docker/gpu/Dockerfile \
     --build-arg="MODEL_ID=$MODEL_ID" \
+    --label="org.opencontainers.image.source=https://github.com/premAI-io/prem-services"
     --tag $IMAGE:latest \
     --tag $IMAGE:$VERSION \
     .