Dockerfile

Dockerfile

@@ -7,9 +7,10 @@ WORKDIR /app
 # Copy the current directory contents into the container at /app
 COPY . /app
 
-# Create a directory for NLTK data and set the environment variable
-RUN mkdir -p /app/nltk_data
+# Create directories for NLTK and Hugging Face cache data and set environment variables
+RUN mkdir -p /app/nltk_data /app/.cache/huggingface
 ENV NLTK_DATA=/app/nltk_data
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
 
 # Install any needed packages specified in requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt