Update Dockerfile

Dockerfile

@@ -2,20 +2,34 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# Install system dependencies (for transformers)
+# Install system dependencies (for transformers and other requirements)
 RUN apt-get update && apt-get install -y \
     git \
     gcc \
+    g++ \
+    make \
     && rm -rf /var/lib/apt/lists/*
 
+# Create cache directory with proper permissions
+RUN mkdir -p /.cache && chmod -R 777 /.cache
+
+# Set environment variables for Hugging Face cache
+ENV TRANSFORMERS_CACHE=/.cache
+ENV HF_HOME=/.cache
+ENV HUGGINGFACE_HUB_CACHE=/.cache
+
 # Install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Optionally pre-download the tokenizer during build
+# Uncomment if you want to download during build (for smaller models)
+# RUN python -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('codellama/CodeLlama-7b-hf')"
+
 # Copy code
 COPY . .
 
-# Expose port (must match Hugging Face's default: 7860)
+# Expose port
 EXPOSE 7860
 
 # Run the API