jjf1999 commited on
Commit
f99c4a4
·
verified ·
1 Parent(s): 3c09dd7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +8 -7
Dockerfile CHANGED
@@ -15,7 +15,7 @@ RUN pip install --no-cache-dir \
15
  torchaudio \
16
  --extra-index-url https://download.pytorch.org/whl/cpu
17
 
18
- # 3. Python 服务与音频处理库
19
  RUN pip install --no-cache-dir \
20
  fastapi \
21
  uvicorn \
@@ -26,9 +26,10 @@ RUN pip install --no-cache-dir \
26
  librosa \
27
  numpy \
28
  scipy \
29
- av
 
30
 
31
- # 4. 创建用户与工作目录
32
  RUN useradd -m -u 1000 user
33
  USER user
34
  ENV HOME=/home/user \
@@ -36,10 +37,10 @@ ENV HOME=/home/user \
36
 
37
  WORKDIR $HOME/app
38
 
39
- # 5. 【关键】:显式下载到固定的本地目录 ./models/
40
- RUN python3 -c "from modelscope import snapshot_download; \
41
- snapshot_download('iic/SenseVoiceSmall', local_dir='/home/user/app/models/SenseVoiceSmall'); \
42
- snapshot_download('iic/emotion2vec_plus_seed', local_dir='/home/user/app/models/emotion2vec_plus_seed')"
43
 
44
  # 6. 复制代码
45
  COPY --chown=user . .
 
15
  torchaudio \
16
  --extra-index-url https://download.pytorch.org/whl/cpu
17
 
18
+ # 3. Python 库
19
  RUN pip install --no-cache-dir \
20
  fastapi \
21
  uvicorn \
 
26
  librosa \
27
  numpy \
28
  scipy \
29
+ av \
30
+ huggingface_hub
31
 
32
+ # 4. 用户与环境
33
  RUN useradd -m -u 1000 user
34
  USER user
35
  ENV HOME=/home/user \
 
37
 
38
  WORKDIR $HOME/app
39
 
40
+ # 5. 【核心改造】:改从 Hugging Face 官方 Hub 下载(走 HF 机房内网,百兆宽带秒级完成)
41
+ RUN python3 -c "from huggingface_hub import snapshot_download; \
42
+ snapshot_download('FunAudioLLM/SenseVoiceSmall', local_dir='/home/user/app/models/SenseVoiceSmall'); \
43
+ snapshot_download('FunAudioLLM/emotion2vec_plus_seed', local_dir='/home/user/app/models/emotion2vec_plus_seed')"
44
 
45
  # 6. 复制代码
46
  COPY --chown=user . .