From 749f5026d65dceaf6fbd10e9612bdc0b3ee3ea35 Mon Sep 17 00:00:00 2001
From: Casper da Costa-Luis <casper.dcl@physics.org>
Date: Thu, 24 Aug 2023 22:12:11 +0100
Subject: [PATCH] label images

- part of #81
---
 utils.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/utils.sh b/utils.sh
index c6e0133..82d10de 100644
--- 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 \
     .