mac9087 commited on
Commit
276589d
·
verified ·
1 Parent(s): ef15b91

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -6,6 +6,7 @@ ENV HF_HOME=/tmp/huggingface
6
  ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
7
  ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
8
  ENV NUMBA_CACHE_DIR=/tmp/numba_cache
 
9
  ENV PYTHONDONTWRITEBYTECODE=1
10
  ENV PYTHONUNBUFFERED=1
11
  ENV CUDA_VISIBLE_DEVICES=""
@@ -27,8 +28,8 @@ RUN apt-get update && apt-get install -y \
27
  && apt-get clean \
28
  && rm -rf /var/lib/apt/lists/*
29
 
30
- RUN mkdir -p /tmp/uploads /tmp/results /tmp/huggingface/transformers /tmp/huggingface/datasets /tmp/numba_cache \
31
- && chmod -R 777 /tmp/uploads /tmp/results /tmp/huggingface /tmp/numba_cache
32
 
33
  # Copy requirements file and install dependencies
34
  COPY requirements.txt .
 
6
  ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
7
  ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
8
  ENV NUMBA_CACHE_DIR=/tmp/numba_cache
9
+ ENV U2NET_HOME=/tmp/u2net
10
  ENV PYTHONDONTWRITEBYTECODE=1
11
  ENV PYTHONUNBUFFERED=1
12
  ENV CUDA_VISIBLE_DEVICES=""
 
28
  && apt-get clean \
29
  && rm -rf /var/lib/apt/lists/*
30
 
31
+ RUN mkdir -p /tmp/uploads /tmp/results /tmp/huggingface/transformers /tmp/huggingface/datasets /tmp/numba_cache /tmp/u2net \
32
+ && chmod -R 777 /tmp/uploads /tmp/results /tmp/huggingface /tmp/numba_cache /tmp/u2net
33
 
34
  # Copy requirements file and install dependencies
35
  COPY requirements.txt .