diff --git a/docker/main/requirements-wheels.txt b/docker/main/requirements-wheels.txt
index 06282147f..624983eb4 100644
--- a/docker/main/requirements-wheels.txt
+++ b/docker/main/requirements-wheels.txt
@@ -11,8 +11,9 @@ joserfc == 1.0.*
 pathvalidate == 3.2.*
 markupsafe == 3.0.*
 python-multipart == 0.0.12
-# Classification Model
-tensorflow-cpu == 2.19.*
+# Classification Model Training
+tensorflow == 2.19.* ; platform_machine == 'aarch64'
+tensorflow-cpu == 2.19.* ; platform_machine == 'x86_64'
 # General
 mypy == 1.6.1
 onvif-zeep-async == 3.1.*