diff --git a/Dockerfile b/Dockerfile index 01cfc0a..ede3b72 100644 --- a/Dockerfile +++ b/Dockerfile @@ -3,7 +3,8 @@ MAINTAINER Andrew Sotheran RUN apt update -y && \ apt install -y python3-pip && \ pip3 install nltk numpy sklearn flask pandas && \ - rm -rf /var/lib/apt/lists/* + rm -rf /var/lib/apt/lists/* && \ COPY . /home/spam-filter/. EXPOSE 9090 +CMD ["python", "/home/spam-filter/configuration/scripts/nltk.py"] CMD ["python", "/home/spam-filter/src/main.py"] \ No newline at end of file diff --git a/configuration/scripts/nltk.py b/configuration/scripts/nltk.py new file mode 100644 index 0000000..df69da2 --- /dev/null +++ b/configuration/scripts/nltk.py @@ -0,0 +1,6 @@ +#!/usr/bin/env python + +import nltk + +nltk.download('pubkt') +nltk.download('stopwords') \ No newline at end of file