diff --git a/machine-learning/Dockerfile b/machine-learning/Dockerfile
index 5b6cc9b6d87c4..ad22e9f3ee3d0 100644
--- a/machine-learning/Dockerfile
+++ b/machine-learning/Dockerfile
@@ -52,7 +52,8 @@ ENV LD_LIBRARY_PATH=/opt/armnn
 RUN apt-get update && apt-get install -y --no-install-recommends ocl-icd-libopencl1 mesa-opencl-icd && \
     rm -rf /var/lib/apt/lists/* && \
     mkdir --parents /etc/OpenCL/vendors && \
-    echo "/usr/lib/libmali.so" > /etc/OpenCL/vendors/mali.icd
+    echo "/usr/lib/libmali.so" > /etc/OpenCL/vendors/mali.icd && \
+    mkdir /opt/armnn
 
 COPY --from=builder-armnn \
       /opt/armnn/libarmnn.so.?? \
diff --git a/machine-learning/app/models/facial_recognition.py b/machine-learning/app/models/facial_recognition.py
index 072fc807f9298..894f5ec7264e4 100644
--- a/machine-learning/app/models/facial_recognition.py
+++ b/machine-learning/app/models/facial_recognition.py
@@ -28,7 +28,10 @@ def __init__(
 
     def _load(self) -> None:
         self.det_model = RetinaFace(session=self._make_session(self.det_file))
-        self.rec_model = ArcFaceONNX(self.rec_file.as_posix(), session=self._make_session(self.rec_file))
+        self.rec_model = ArcFaceONNX(
+            self.rec_file.with_suffix(".onnx").as_posix(),
+            session=self._make_session(self.rec_file),
+        )
 
         self.det_model.prepare(
             ctx_id=0,