jisaacso219 commited on
Commit
935dc92
·
verified ·
1 Parent(s): c0b35a6

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -5
Dockerfile CHANGED
@@ -1,17 +1,16 @@
1
  FROM python:3.10-slim
2
 
3
- # Fix numba cache crash
4
  ENV NUMBA_CACHE_DIR=/tmp/numba_cache
 
5
 
6
  RUN apt-get update && apt-get install -y git ffmpeg libsndfile1 && apt-get clean
7
-
8
  WORKDIR /app
9
  COPY . /app
10
 
11
  RUN pip install --upgrade pip
12
-
13
- # Install all packages from PyTorch + working combo
14
- RUN pip install --extra-index-url https://download.pytorch.org/whl/cpu -r requirements.txt
15
 
16
  EXPOSE 7860
17
  CMD ["python", "app.py"]
 
1
  FROM python:3.10-slim
2
 
 
3
  ENV NUMBA_CACHE_DIR=/tmp/numba_cache
4
+ ENV NUMBA_DISABLE_CACHE=1
5
 
6
  RUN apt-get update && apt-get install -y git ffmpeg libsndfile1 && apt-get clean
 
7
  WORKDIR /app
8
  COPY . /app
9
 
10
  RUN pip install --upgrade pip
11
+ RUN pip install --extra-index-url https://download.pytorch.org/whl/cpu torch==2.2.2+cpu torchaudio==2.2.2+cpu \
12
+ https://download.pytorch.org/whl/cpu/torchvision-0.17.2%2Bcpu-cp310-cp310-linux_x86_64.whl
13
+ RUN pip install TTS==0.22.0 transformers flask sentencepiece numpy==1.22.0 scipy librosa==0.10.1 numba==0.57.1
14
 
15
  EXPOSE 7860
16
  CMD ["python", "app.py"]