use different setup

Dockerfile

@@ -1,15 +1,32 @@
-FROM python:3.10-slim
+FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
+
+# Set environment variables
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
+ENV LANG=C.UTF-8
 
 # Set working directory
 WORKDIR /app
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
+    python3.10 \
+    python3.10-dev \
+    python3-pip \
     git \
     build-essential \
+    cmake \
+    gcc \
+    g++ \
     libsndfile1 \
+    sox \
+    libsox-dev \
+    libsox-fmt-all \
     && rm -rf /var/lib/apt/lists/*
 
+# Upgrade pip
+RUN python3.10 -m pip install --upgrade pip setuptools wheel
+
 # Copy requirements first to leverage Docker cache
 COPY requirements.txt .
 
@@ -19,6 +36,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Clone and install NeMo
 RUN git clone https://github.com/AI4Bharat/NeMo.git && \
     cd NeMo && \
+    git checkout nemo-v2 && \
     pip install -e .
 
 # Copy application code

requirements.txt

@@ -1,4 +1,3 @@
-# requirements.txt
 fastapi==0.104.1
 uvicorn==0.24.0
 python-multipart==0.0.6
@@ -8,18 +7,6 @@ torchaudio==2.1.0
 torchvision==0.16.0
 packaging==23.2
 huggingface_hub==0.23.2
-numpy>=1.20.0
+numpy>=1.20.0,<1.24.0
 soundfile>=0.12.1
 librosa>=0.10.1
-omegaconf>=2.3.0
-hydra-core>=1.3.2
-pytorch-lightning>=2.1.0
-webdataset>=0.1.62
-transformers>=4.36.0
-sacremoses>=0.0.53
-youtokentome>=1.0.6
-numpy<1.24.0
-einops>=0.6.1
-contextlib2>=21.6.0
-inflect>=7.0.0
-typing_extensions>=4.8.0