rcastriotta commited on
Commit
d2ed72d
·
verified ·
1 Parent(s): 3395e3f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -4
Dockerfile CHANGED
@@ -1,5 +1,6 @@
1
  FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
2
 
 
3
  ENV DEBIAN_FRONTEND=noninteractive
4
 
5
  RUN apt-get update && apt-get install -y \
@@ -12,13 +13,11 @@ RUN apt-get update && apt-get install -y \
12
  RUN --mount=type=cache,target=/root/.cache \
13
  pip3 install "audio-separator[gpu]" fastapi uvicorn python-multipart ffmpeg-python pydantic youtube-transcript-api
14
 
15
- COPY . .
16
 
17
  EXPOSE 7860
18
 
19
- ENV TMP_DIRECTORY=tmp
20
-
21
  RUN mkdir -p ${TMP_DIRECTORY}
22
-
23
 
24
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
 
1
  FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
2
 
3
+ WORKDIR /app
4
  ENV DEBIAN_FRONTEND=noninteractive
5
 
6
  RUN apt-get update && apt-get install -y \
 
13
  RUN --mount=type=cache,target=/root/.cache \
14
  pip3 install "audio-separator[gpu]" fastapi uvicorn python-multipart ffmpeg-python pydantic youtube-transcript-api
15
 
16
+ COPY . /app
17
 
18
  EXPOSE 7860
19
 
 
 
20
  RUN mkdir -p ${TMP_DIRECTORY}
21
+ ENV NUMBA_CACHE_DIR=/tmp
22
 
23
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]