Dockerfile

Dockerfile

@@ -8,9 +8,11 @@ WORKDIR /app
 COPY . /app
 
 # Create directories for NLTK and Hugging Face cache data and set environment variables
-RUN mkdir -p /app/nltk_data /app/.cache/huggingface
+RUN mkdir -p /app/nltk_data /app/.cache/huggingface /app/.huggingface_cache \
+    && chown -R 1000:1000 /app/nltk_data /app/.cache/huggingface /app/.huggingface_cache
+
 ENV NLTK_DATA=/app/nltk_data
-ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
+ENV HF_HOME=/app/.huggingface_cache
 
 # Install any needed packages specified in requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt