umgefahren
commited on
Commit
·
216dc23
1
Parent(s):
6064b6f
Add cache
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
|
@@ -28,6 +28,8 @@ RUN python -m pip install gradio
|
|
| 28 |
|
| 29 |
COPY . .
|
| 30 |
|
| 31 |
-
|
|
|
|
|
|
|
| 32 |
|
| 33 |
CMD ["python", "main.py"]
|
|
|
|
| 28 |
|
| 29 |
COPY . .
|
| 30 |
|
| 31 |
+
ENV TORCH_HOME /data/torch
|
| 32 |
+
ENV MPLCONFIGDIR /data/matplotlib
|
| 33 |
+
|
| 34 |
|
| 35 |
CMD ["python", "main.py"]
|