Update Dockerfile

Dockerfile

@@ -1,32 +1,36 @@
-# Use official slim Python 3.12 image
+# Use a slim Python 3.12 base image
 FROM python:3.12-slim
 
-# Set working directory
+# Set environment variables to suppress warnings and ensure non-interactive installs
+ENV PYTHONUNBUFFERED=1 \
+    PIP_NO_CACHE_DIR=1 \
+    TRANSFORMERS_CACHE=/tmp/transformers_cache \
+    HF_HOME=/tmp/huggingface_cache
+
+# Set the working directory inside the container
 WORKDIR /app
 
-# Copy app code
+# Copy the application code into the container
 COPY app.py .
 
-# Set environment variables for writable caches
-ENV HF_HOME=/tmp/hf_cache
-ENV TRANSFORMERS_CACHE=/tmp/hf_cache
-ENV PIP_CACHE_DIR=/tmp/pip_cache
-ENV PIP_NO_CACHE_DIR=off
-
-# Upgrade pip
-RUN python -m pip install --upgrade pip
-
-# Install dependencies
-# Remove en-core-web-sm from pip, install via spacy download if needed at runtime
-RUN pip install --no-cache-dir \
-    torch \
+# Install system dependencies and Python packages
+RUN apt-get update && apt-get install -y \
+    espeak-ng \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/* \
+    && python -m pip install --upgrade pip \
+    && pip install --no-cache-dir \
+    torch==2.9.0 \
     soundfile \
     fastapi \
     uvicorn \
-    kokoro
+    spacy \
+    && python -m spacy download en_core_web_sm==3.8.0 \
+    && pip install --no-cache-dir \
+    kokoro==0.9.4
 
-# Expose FastAPI port
-EXPOSE 7860
+# Expose the port the app runs on
+EXPOSE 7860 
 
-# Launch FastAPI app
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Command to run the application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]