Added option to download NLTK data sets for non Docker users

Download missing nltk datasets in setup.py
ukanth · Feb 28, 2017 · ed6b9fc · ed6b9fc
2 parents 93030fc + 262b24a
commit ed6b9fc
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -1,6 +1,12 @@
 import nltk
 import os
 
+# Downloading necessary NLTK datasets
+nltk.download("stopwords")
+nltk.download("wordnet")
+nltk.download('averaged_perceptron_tagger')
+nltk.download('punkt')
+
 #creating directory for storing chat logs
 if not os.path.exists("logs"):
     os.makedirs("logs")
@@ -43,4 +49,4 @@
     newStory.labeledSentences.append(newLabeledSentence)
     newStory.save()
 except:
-    print("Stories already exists..skipping..")
+    print("Stories already exists..skipping..")