diff --git a/minimal-ml-inference/Dockerfile.model b/minimal-ml-inference/Dockerfile.model
index d0380a3..5bde9c2 100644
--- a/minimal-ml-inference/Dockerfile.model
+++ b/minimal-ml-inference/Dockerfile.model
@@ -1,8 +1,8 @@
 # syntax=docker/dockerfile:1
 
-ARG UBUNTU_VERSION=22.04
+ARG DEVICE_PLATFORM=linux/arm64/v8
 
-FROM arm64v8/alpine as model-image
+FROM --platform=$DEVICE_PLATFORM alpine as model-image
 
 # Get SSD Mobilenet V2
 ADD https://github.com/google-coral/edgetpu/raw/master/test_data/ssd_mobilenet_v2_coco_quant_postprocess.tflite models/
diff --git a/object-detector-python/Dockerfile.model b/object-detector-python/Dockerfile.model
index 192b556..5bde9c2 100644
--- a/object-detector-python/Dockerfile.model
+++ b/object-detector-python/Dockerfile.model
@@ -1,6 +1,8 @@
 # syntax=docker/dockerfile:1
 
-FROM arm64v8/alpine as model-image
+ARG DEVICE_PLATFORM=linux/arm64/v8
+
+FROM --platform=$DEVICE_PLATFORM alpine as model-image
 
 # Get SSD Mobilenet V2
 ADD https://github.com/google-coral/edgetpu/raw/master/test_data/ssd_mobilenet_v2_coco_quant_postprocess.tflite models/
diff --git a/pose-estimator-with-flask/Dockerfile.model b/pose-estimator-with-flask/Dockerfile.model
index d922f5d..82fce41 100644
--- a/pose-estimator-with-flask/Dockerfile.model
+++ b/pose-estimator-with-flask/Dockerfile.model
@@ -1,6 +1,8 @@
 # syntax=docker/dockerfile:1
 
-FROM arm64v8/alpine
+ARG DEVICE_PLATFORM=linux/arm64/v8
+
+FROM --platform=$DEVICE_PLATFORM alpine as model-image
 
 # Get SSD Mobilenet V2
 ADD https://raw.githubusercontent.com/google-coral/test_data/master/movenet_single_pose_lightning_ptq_edgetpu.tflite models/