pvanand commited on
Commit
d59fbd1
1 Parent(s): b0f97ff

Update download_nltk_data.py

Browse files
Files changed (1) hide show
  1. download_nltk_data.py +3 -3
download_nltk_data.py CHANGED
@@ -1,9 +1,9 @@
1
  import nltk
2
  import os
3
 
4
- # Set the NLTK_DATA environment variable
5
  os.environ['NLTK_DATA'] = '/app/nltk_data'
6
 
7
  # Download NLTK data
8
- nltk.download('stopwords')
9
- nltk.download('punkt')
 
1
  import nltk
2
  import os
3
 
4
+ # Set the NLTK_DATA environment variable to /app/nltk_data
5
  os.environ['NLTK_DATA'] = '/app/nltk_data'
6
 
7
  # Download NLTK data
8
+ nltk.download('stopwords', download_dir=os.environ['NLTK_DATA'])
9
+ nltk.download('punkt', download_dir=os.environ['NLTK_DATA'])