diff --git a/app.py b/app.py
index 1807a80..71285f0 100644
--- a/app.py
+++ b/app.py
@@ -144,6 +144,10 @@ def __init__(self, scope: Construct, id: str, **kwargs) -> None:
             task_role=role,
             cpu=4 * 1024,
             memory_limit_mib=8 * 1024,
+            runtime_platform=ecs.RuntimePlatform(
+                operating_system_family=ecs.OperatingSystemFamily.LINUX,
+                cpu_architecture=ecs.CpuArchitecture.ARM64
+            ),
         )
 
         container = task_definition.add_container(
diff --git a/container/Dockerfile b/container/Dockerfile
index 72a3817..a17f159 100644
--- a/container/Dockerfile
+++ b/container/Dockerfile
@@ -1,4 +1,4 @@
-FROM python:3.10.12
+FROM --platform=linux/amd64 python:3.10.12
 
 RUN pip install \
     mlflow==2.6.0 \