feat: add custom layer

petercat-ai · Apr 15, 2024 · 0adc42b · 0adc42b
1 parent 2b9c437
commit 0adc42b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/server/Dockerfile.aws.lambda b/server/Dockerfile.aws.lambda
@@ -8,4 +8,13 @@ COPY . ${LAMBDA_TASK_ROOT}
 COPY requirements.txt .
 RUN pip3 install -r requirements.txt --target "${LAMBDA_TASK_ROOT}" -U --no-cache-dir
 
+# Setup NLTK again in system path to execute nltk.downloader
+RUN pip install nltk
+# Setup directory for NLTK_DATA
+RUN mkdir -p ./nltk_data
+
+# Download NLTK_DATA to build directory
+RUN python -W ignore -m nltk.downloader punkt -d ./nltk_data
+RUN python -W ignore -m nltk.downloader stopwords -d ./nltk_data
+
 CMD ["python", "main.py"]