Update Dockerfile

Dockerfile

@@ -1,5 +1,9 @@
 FROM python:3.9-slim
 
+# Create cache directory with proper permissions
+RUN mkdir -p /app/.cache && \
+    chmod 777 /app/.cache
+
 # Install system dependencies
 RUN apt-get update && \
     apt-get install -y \
@@ -13,13 +17,18 @@ RUN apt-get update && \
 
 WORKDIR /app
 
+# Set environment variables for Hugging Face
+ENV TRANSFORMERS_CACHE=/app/.cache \
+    HF_DATASETS_CACHE=/app/.cache \
+    XDG_CACHE_HOME=/app/.cache
+
 # Install Python dependencies
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt --mkdir
 
-# Test model download during build
+# Test model download with proper cache
 RUN python -c "from transformers import GPT2LMHeadModel; \
-GPT2LMHeadModel.from_pretrained('gpt2-medium', from_tf=False, use_safetensors=True)"
+GPT2LMHeadModel.from_pretrained('gpt2-medium', local_files_only=False)"
 
 COPY . .