Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -12,10 +12,10 @@ RUN apt-get update && apt-get install -y \
|
|
| 12 |
&& rm -rf /var/lib/apt/lists/*
|
| 13 |
|
| 14 |
# Create writable cache directories within /tmp
|
| 15 |
-
RUN mkdir -p /tmp/
|
| 16 |
|
| 17 |
# Set environment variables for Hugging Face Transformers and Matplotlib
|
| 18 |
-
ENV
|
| 19 |
ENV MPLCONFIGDIR=/tmp/matplotlib
|
| 20 |
|
| 21 |
# Copy requirements.txt first to leverage Docker caching
|
|
|
|
| 12 |
&& rm -rf /var/lib/apt/lists/*
|
| 13 |
|
| 14 |
# Create writable cache directories within /tmp
|
| 15 |
+
RUN mkdir -p /tmp/huggingface_cache /tmp/matplotlib
|
| 16 |
|
| 17 |
# Set environment variables for Hugging Face Transformers and Matplotlib
|
| 18 |
+
ENV HF_HOME=/tmp/huggingface_cache
|
| 19 |
ENV MPLCONFIGDIR=/tmp/matplotlib
|
| 20 |
|
| 21 |
# Copy requirements.txt first to leverage Docker caching
|