changes

Dockerfile

@@ -1,21 +1,41 @@
 FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-devel
+
 ARG DEBIAN_FRONTEND=noninteractive
 
+# Install system dependencies
 RUN apt-get update && \
-    apt-get install --no-install-recommends -y sox libsox-fmt-all curl wget gcc git git-lfs build-essential libaio-dev libsndfile1 ssh ffmpeg && \
+    apt-get install -y --no-install-recommends \
+    sox libsox-fmt-all curl wget gcc git git-lfs \
+    libsndfile1 ffmpeg build-essential && \
     apt-get clean && apt-get -y autoremove
 
 WORKDIR /app
+
+# Upgrade pip
+RUN pip install --upgrade pip
+
+# Install Python dependencies
 COPY requirements.txt .
-RUN python -m pip install --use-deprecated=legacy-resolver -r requirements.txt \
-    && python -m pip cache purge
+RUN pip install -r requirements.txt && \
+    pip cache purge
 
+# Download unidic dictionary
 RUN python -m unidic download
+
+# Create directory for TTS models
 RUN mkdir -p /app/tts_models
 
+# Copy application code
 COPY app.py .
-ENV NVIDIA_DISABLE_REQUIRE=1
 
+# Set environment variables
+ENV NUMBA_DISABLE_CACHE=1
+ENV NVIDIA_DISABLE_REQUIRE=1
 ENV NUM_THREADS=2
+ENV USE_CPU=0  
+
+# Expose the correct port (match with test_streaming.py)
 EXPOSE 7860
+
+# Run the Uvicorn server on the correct port
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

server/requirements.txt

@@ -1,12 +1,17 @@
-TTS @ git+https://github.com/coqui-ai/TTS@fa28f99f1508b5b5366539b2149963edcb80ba62
-uvicorn[standard]==0.23.2
-fastapi==0.95.2
-deepspeed
-pydantic==1.10.13
+TTS @ git+https://github.com/coqui-ai/TTS.git@fa28f99f1508b5b5366539b2149963edcb80ba62
+pyannote==1.0.13
 python-multipart==0.0.6
-typing-extensions>=4.8.0
-numpy
-cutlet
+typing-extensions==4.8.0
+cutlet==0.3.0
 mecab-python3==1.0.6
 unidic-lite==1.0.8
 unidic==1.1.0
+langid==1.1.6
+deepspeed==0.14.5
+pydub==0.25.1
+uvicorn==0.30.6
+fastapi==0.115.0
+numpy==1.24.4
+scipy==1.13.1
+numba==0.58.1
+librosa==0.10.2