yash184 commited on
Commit
e48ec1b
·
verified ·
1 Parent(s): fb657a8

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -4
Dockerfile CHANGED
@@ -13,12 +13,11 @@ WORKDIR /home/user/app
13
  RUN pip install --no-cache-dir --upgrade pip && \
14
  pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu
15
 
16
- # Hindi model ko "append_terminal_punctuation: false" support chahiye, isliye GitHub ka latest version
17
  RUN pip install --no-cache-dir "git+https://github.com/kyutai-labs/pocket-tts.git" \
18
- fastapi "uvicorn[standard]" numpy
19
 
20
- # Model build ke time hi download kar lo (fail ho to bhi build chalta rahe)
21
- RUN python -c "from pocket_tts import TTSModel; TTSModel.load_model('hf://saryps-labs/pocket-tts-hindi/config.yaml')" || true
22
 
23
  COPY --chown=user . /home/user/app
24
 
 
13
  RUN pip install --no-cache-dir --upgrade pip && \
14
  pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu
15
 
 
16
  RUN pip install --no-cache-dir "git+https://github.com/kyutai-labs/pocket-tts.git" \
17
+ fastapi "uvicorn[standard]" numpy huggingface_hub
18
 
19
+ # Hindi model repo ko local folder me download karo (config.yaml local path chahiye)
20
+ RUN python -c "from huggingface_hub import snapshot_download; snapshot_download('saryps-labs/pocket-tts-hindi', local_dir='/home/user/hindi_model')" || true
21
 
22
  COPY --chown=user . /home/user/app
23