diff --git a/Dockerfile b/Dockerfile
index 853dd5d..7c66584 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -19,6 +19,7 @@ WORKDIR /app
 COPY package.json ./
 
 RUN npm i
+RUN npm rebuild @tensorflow/tfjs-node --build-from-source
 
 EXPOSE 7272