Update Dockerfile

Dockerfile

@@ -79,11 +79,14 @@ COPY requirements.txt .
 RUN pip install --upgrade pip
 RUN pip install --no-cache-dir --verbose -r requirements.txt
 
-# Set a writable cache directory
-ENV TRANSFORMERS_CACHE=/app/cache
+# Set a writable cache directory for Hugging Face Transformers
 ENV HF_HOME=/app/cache
 RUN mkdir -p /app/cache && chmod -R 777 /app/cache
 
+# Set a writable cache directory for Numba
+ENV NUMBA_CACHE_DIR=/app/cache/numba_cache
+RUN mkdir -p /app/cache/numba_cache && chmod -R 777 /app/cache/numba_cache
+
 # Set writable Matplotlib config directory
 ENV MPLCONFIGDIR=/app/cache/matplotlib
 RUN mkdir -p $MPLCONFIGDIR && chmod -R 777 $MPLCONFIGDIR