Update Dockerfile

Dockerfile

@@ -82,13 +82,14 @@ RUN pip install --no-cache-dir --verbose -r requirements.txt
 # Set a writable cache directory
 ENV TRANSFORMERS_CACHE=/app/cache
 
-# Disable Numba caching
-ENV NUMBA_CACHE_DIR=/app/numba_cache
-RUN mkdir -p /app/numba_cache
+ENV NUMBA_DISABLE_JIT=1
 
-# Ensure the cache directory exists
+
+# Set writable cache directory using HF_HOME
+ENV HF_HOME=/app/cache
 RUN mkdir -p /app/cache
 
+
 # Download NLTK data
 RUN python -m nltk.downloader punkt