Update Dockerfile

Dockerfile

@@ -1,5 +1,9 @@
 FROM python:3.9-slim
 
+# Set environment variables for cache directories
+ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
+ENV HF_HOME=/tmp/hf_home
+
 WORKDIR /app
 
 RUN apt-get update && apt-get install -y \
@@ -9,6 +13,9 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Create cache directories
+RUN mkdir -p /tmp/transformers_cache /tmp/hf_home
+
 COPY requirements.txt ./
 COPY src/ ./src/