Update Dockerfile

Dockerfile

@@ -11,6 +11,13 @@ RUN apt-get update && apt-get install -y \
     libomp-dev \
     && rm -rf /var/lib/apt/lists/*
 
+# Create writable cache directories within /tmp
+RUN mkdir -p /tmp/transformers_cache /tmp/matplotlib
+
+# Set environment variables for Hugging Face Transformers and Matplotlib
+ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
+ENV MPLCONFIGDIR=/tmp/matplotlib
+
 # Copy requirements.txt first to leverage Docker caching
 COPY requirements.txt /app/requirements.txt
 
@@ -20,9 +27,12 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Install PyTorch separately to ensure compatibility with the correct architecture
 RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
 
-# Copy the rest of the application code, including the model file
+# Copy the rest of the application code, including the local model file
 COPY . /app
 
+# Ensure that bert_imdb_model.bin exists
+RUN if [ ! -f "/app/bert_imdb_model.bin" ]; then echo "Error: Model file not found in container"; exit 1; fi
+
 # Expose port 7860 for Hugging Face Spaces
 EXPOSE 7860