Set TRANSFORMERS_CACHE environment variable to a writable directory

Dockerfile

@@ -1,10 +1,17 @@
-FROM python:3.9-slim
+FROM python:3.9
 
+# Set the TRANSFORMERS_CACHE environment variable
+ENV TRANSFORMERS_CACHE=/app/cache/huggingface/hub
+
+# Copy your app code
+COPY . /app
 WORKDIR /app
 
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+# Install dependencies
+RUN pip install -r requirements.txt
 
-COPY . .
+# Create the cache directory
+RUN mkdir -p /app/cache/huggingface/hub
 
-CMD ["python", "app.py"]
+# Start the app
+CMD ["python", "app.py"]

start.sh

@@ -0,0 +1,10 @@
+#!/bin/bash
+
+# Create the cache directory if it doesn't exist
+mkdir -p ./cache/huggingface/hub
+
+# Set the TRANSFORMERS_CACHE environment variable
+export TRANSFORMERS_CACHE=$(pwd)/cache/huggingface/hub
+
+# Start the app
+python app.py