Fix pie dockerfile

PrinsINT · PrinsINT · commit e064e54f979c · 2025-04-02T17:08:59.000+02:00
diff --git a/pie/TDN-1200-1600/Dockerfile b/pie/TDN-1200-1600/Dockerfile
@@ -1,4 +1,6 @@
 ARG VERSION=dev
-FROM instituutnederlandsetaal/galahad-taggers-pie-base:$VERSION
+ARG CPU_GPU=${CPU_GPU}
+
+FROM instituutnederlandsetaal/galahad-taggers-pie-$CPU_GPU-base:$VERSION
 
 COPY --link *.tar ./model.tar
diff --git a/pie/TDN-1600-1900/Dockerfile b/pie/TDN-1600-1900/Dockerfile
@@ -1,4 +1,6 @@
 ARG VERSION=dev
-FROM instituutnederlandsetaal/galahad-taggers-pie-base:$VERSION
+ARG CPU_GPU=${CPU_GPU}
+
+FROM instituutnederlandsetaal/galahad-taggers-pie-$CPU_GPU-base:$VERSION
 
 COPY --link *.tar ./model.tar
diff --git a/pie/TDN-ALL/Dockerfile b/pie/TDN-ALL/Dockerfile
@@ -1,4 +1,6 @@
 ARG VERSION=dev
-FROM instituutnederlandsetaal/galahad-taggers-pie-base:$VERSION
+ARG CPU_GPU=${CPU_GPU}
+
+FROM instituutnederlandsetaal/galahad-taggers-pie-$CPU_GPU-base:$VERSION
 
 COPY --link *.tar ./model.tar
