diff --git a/Dockerfile b/Dockerfile
index fdae322..56fa5f7 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -21,6 +21,16 @@ RUN python3 -m pip install -e .
 RUN python3 -m pip install -e incl/pyinfra
 RUN python3 -m pip install -e incl/pdf2image
 
+# User for the process to run as
+RUN groupadd --gid 1001 non-root \
+ && useradd --uid 1001 --gid 1001 -m non-root
+
+COPY --chown=1001:1001 ./src/serve.py ./src/serve.py
+RUN chown -R 1001:1001 ./image_prediction ./image_prediction
+RUN chown -R 1001:1001 ./config.yaml ./config.yaml
+RUN chown -R 1001:1001 ./banner.txt ./banner.txt
+RUN chown -R 1001:1001 ./incl ./incl
+
 EXPOSE 5000
 EXPOSE 8080