audio_chat / download_nltk_data.py
pvanand's picture
Update download_nltk_data.py
d59fbd1 verified
raw
history blame
273 Bytes
import nltk
import os
# Set the NLTK_DATA environment variable to /app/nltk_data
os.environ['NLTK_DATA'] = '/app/nltk_data'
# Download NLTK data
nltk.download('stopwords', download_dir=os.environ['NLTK_DATA'])
nltk.download('punkt', download_dir=os.environ['NLTK_DATA'])