use different setup

Dockerfile

@@ -37,10 +37,11 @@ RUN pip install --no-cache-dir -r requirements.txt
 RUN git clone https://github.com/AI4Bharat/NeMo.git && \
     cd NeMo && \
     git checkout nemo-v2 && \
-    pip install -e .
+    pip install -e . && \
+    python3.10 -c "import nltk; nltk.download('punkt')"
 
 # Copy application code
-COPY app.py .
+COPY main.py ./main.py
 
 # Create directory for temporary files
 RUN mkdir -p /tmp/audio_files
@@ -49,4 +50,4 @@ RUN mkdir -p /tmp/audio_files
 EXPOSE 8000
 
 # Command to run the application
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]

requirements.txt

@@ -10,3 +10,9 @@ huggingface_hub==0.23.2
 numpy>=1.20.0,<1.24.0
 soundfile>=0.12.1
 librosa>=0.10.1
+hydra-core>=1.3.2
+omegaconf>=2.3.0
+pytorch-lightning>=2.1.0
+transformers>=4.36.0
+sacremoses>=0.0.53
+nltk>=3.8.1