TonyLikeDev commited on
Commit
b16f2bd
·
1 Parent(s): bd724a8

Fix no df

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -5,6 +5,7 @@ WORKDIR /app
5
  RUN apt-get update && apt-get install -y ffmpeg && rm -rf /var/lib/apt/lists/*
6
 
7
  COPY requirements.txt .
 
8
  RUN pip install --no-cache-dir -r requirements.txt
9
 
10
  RUN python -c "
@@ -15,7 +16,7 @@ for d in site.getsitepackages():
15
  src = p.read_text()
16
  src = src.replace(
17
  'from torchaudio.backend.common import AudioMetaData',
18
- 'try:\n from torchaudio.backend.common import AudioMetaData\nexcept ImportError:\n from torchaudio import AudioMetaData'
19
  )
20
  p.write_text(src)
21
  break
 
5
  RUN apt-get update && apt-get install -y ffmpeg && rm -rf /var/lib/apt/lists/*
6
 
7
  COPY requirements.txt .
8
+ RUN pip install --no-cache-dir torch==2.2.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cpu
9
  RUN pip install --no-cache-dir -r requirements.txt
10
 
11
  RUN python -c "
 
16
  src = p.read_text()
17
  src = src.replace(
18
  'from torchaudio.backend.common import AudioMetaData',
19
+ 'try:\n from torchaudio.backend.common import AudioMetaData\nexcept ImportError:\n from torchaudio._backend.common import AudioMetaData'
20
  )
21
  p.write_text(src)
22
  break