Pushkar02-n commited on
Commit
75d6bee
·
verified ·
1 Parent(s): 0b0dcfb

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -4
Dockerfile CHANGED
@@ -2,13 +2,12 @@ FROM python:3.10-slim
2
 
3
  WORKDIR /app
4
 
5
- # Install system dependencies (added ffmpeg just in case torchaudio needs it for your audio types)
6
  RUN apt-get update && apt-get install -y \
7
  git bash wget build-essential libsndfile1 ffmpeg \
8
  && rm -rf /var/lib/apt/lists/*
9
 
10
  # Combine the NeMo install and the PyTorch CPU override into ONE step.
11
- # This ensures matching versions (2.1.2) and prevents Docker from caching massive GPU binaries.
12
  RUN git clone https://github.com/AI4Bharat/NeMo.git \
13
  && cd NeMo \
14
  && git checkout nemo-v2 \
@@ -17,8 +16,8 @@ RUN git clone https://github.com/AI4Bharat/NeMo.git \
17
  && pip install --no-cache-dir torch==2.1.2+cpu torchvision==0.16.2+cpu torchaudio==2.1.2+cpu --index-url https://download.pytorch.org/whl/cpu \
18
  && cd ..
19
 
20
- # Install FastAPI, Piper TTS, and patch the known dependency issues
21
- RUN pip install --no-cache-dir "numpy<2.0" huggingface_hub==0.23.2 fastapi uvicorn python-multipart "piper-tts==1.2.0"
22
 
23
  # Fetch the Piper Nepali Chitwan model directly
24
  RUN wget -q -O chitwan.onnx "https://huggingface.co/rhasspy/piper-voices/resolve/main/ne/ne_NP/chitwan/medium/ne_NP-chitwan-medium.onnx?download=true"
 
2
 
3
  WORKDIR /app
4
 
5
+ # Install system dependencies
6
  RUN apt-get update && apt-get install -y \
7
  git bash wget build-essential libsndfile1 ffmpeg \
8
  && rm -rf /var/lib/apt/lists/*
9
 
10
  # Combine the NeMo install and the PyTorch CPU override into ONE step.
 
11
  RUN git clone https://github.com/AI4Bharat/NeMo.git \
12
  && cd NeMo \
13
  && git checkout nemo-v2 \
 
16
  && pip install --no-cache-dir torch==2.1.2+cpu torchvision==0.16.2+cpu torchaudio==2.1.2+cpu --index-url https://download.pytorch.org/whl/cpu \
17
  && cd ..
18
 
19
+ # Install FastAPI, Piper TTS, and patch the known dependency issues (Added pyarrow downgrade here)
20
+ RUN pip install --no-cache-dir "numpy<2.0" "pyarrow<15.0.0" huggingface_hub==0.23.2 fastapi uvicorn python-multipart "piper-tts==1.2.0"
21
 
22
  # Fetch the Piper Nepali Chitwan model directly
23
  RUN wget -q -O chitwan.onnx "https://huggingface.co/rhasspy/piper-voices/resolve/main/ne/ne_NP/chitwan/medium/ne_NP-chitwan-medium.onnx?download=true"