Update Dockerfile
Browse files- Dockerfile +9 -6
Dockerfile
CHANGED
|
@@ -30,9 +30,15 @@ RUN pip install --no-cache-dir \
|
|
| 30 |
torchcodec==0.6.0 \
|
| 31 |
--index-url https://download.pytorch.org/whl/cu128
|
| 32 |
|
| 33 |
-
#
|
| 34 |
-
|
| 35 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
|
| 37 |
# Install NeMo toolkit with ASR extras
|
| 38 |
RUN pip install --no-cache-dir \
|
|
@@ -53,9 +59,6 @@ ENV HF_AUDIO_DECODER_BACKEND=soundfile
|
|
| 53 |
# Copy the full repository
|
| 54 |
COPY . /app
|
| 55 |
|
| 56 |
-
# Default working directory for running nemo scripts
|
| 57 |
-
WORKDIR /app/nemo_asr
|
| 58 |
-
|
| 59 |
# Default entrypoint
|
| 60 |
ENTRYPOINT ["bash"]
|
| 61 |
|
|
|
|
| 30 |
torchcodec==0.6.0 \
|
| 31 |
--index-url https://download.pytorch.org/whl/cu128
|
| 32 |
|
| 33 |
+
# Install common requirements (torch already installed above, pip will skip it)
|
| 34 |
+
RUN pip install --no-cache-dir \
|
| 35 |
+
transformers \
|
| 36 |
+
evaluate \
|
| 37 |
+
datasets \
|
| 38 |
+
librosa \
|
| 39 |
+
jiwer \
|
| 40 |
+
num2words \
|
| 41 |
+
peft
|
| 42 |
|
| 43 |
# Install NeMo toolkit with ASR extras
|
| 44 |
RUN pip install --no-cache-dir \
|
|
|
|
| 59 |
# Copy the full repository
|
| 60 |
COPY . /app
|
| 61 |
|
|
|
|
|
|
|
|
|
|
| 62 |
# Default entrypoint
|
| 63 |
ENTRYPOINT ["bash"]
|
| 64 |
|