Update Dockerfile
Browse files- Dockerfile +5 -0
Dockerfile
CHANGED
|
@@ -9,6 +9,11 @@ ENV PYTHONDONTWRITEBYTECODE 1
|
|
| 9 |
# Ensure Python output is sent straight to terminal (useful for logs)
|
| 10 |
ENV PYTHONUNBUFFERED 1
|
| 11 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
# Upgrade pip
|
| 13 |
RUN python -m pip install --upgrade pip
|
| 14 |
|
|
|
|
| 9 |
# Ensure Python output is sent straight to terminal (useful for logs)
|
| 10 |
ENV PYTHONUNBUFFERED 1
|
| 11 |
|
| 12 |
+
|
| 13 |
+
# Set environment variables for the cache
|
| 14 |
+
ENV HF_HOME="/app/cache"
|
| 15 |
+
ENV TORCH_HOME="/app/cache"
|
| 16 |
+
|
| 17 |
# Upgrade pip
|
| 18 |
RUN python -m pip install --upgrade pip
|
| 19 |
|