diff --git a/api/Dockerfile b/api/Dockerfile
index d4c18a7..79fd1de 100644
--- a/api/Dockerfile
+++ b/api/Dockerfile
@@ -14,6 +14,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 COPY src .
 COPY start_server.sh .
+COPY build_models .
 
 RUN chmod +x start_server.sh