From 76f069d4727f5001d3b1708f11134a727b7d0798 Mon Sep 17 00:00:00 2001
From: mammo0 <marc.ammon@hotmail.de>
Date: Thu, 11 Mar 2021 07:21:33 +0100
Subject: [PATCH] add an option to include the language models into the Docker
 image

-> this allows the direct offline usage of the Docker image
---
 Dockerfile | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 2452638..661bf92 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,11 +1,21 @@
 FROM python:3.8
 
+ARG with_models=false
+
 WORKDIR /app
 
 RUN pip install --upgrade pip
 
 COPY . .
 
+# check for offline build
+RUN if [ "$with_models" = "true" ]; then  \
+        # install only the dependencies first
+        pip install -e .;  \
+        # initialize the language models
+        ./install_models.py;  \
+    fi
+
 # Install package from source code
 RUN pip install .