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'])