Update Dockerfile

Dockerfile

@@ -1,16 +1,28 @@
+# Use a lightweight base image with Python
 FROM python:3.10-slim
 
-ENV NUMBA_CACHE_DIR=/tmp/numba_cache
-ENV NUMBA_DISABLE_CACHE=1
+# Set environment variables
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
 
-RUN apt-get update && apt-get install -y git ffmpeg libsndfile1 && apt-get clean
+# Install required system packages
+RUN apt-get update && apt-get install -y \
+    ffmpeg \
+ && apt-get clean \
+ && rm -rf /var/lib/apt/lists/*
+
+# Set the working directory
 WORKDIR /app
+
+# Copy app files
 COPY . /app
 
-RUN pip install --upgrade pip
-RUN pip install --extra-index-url https://download.pytorch.org/whl/cpu torch==2.2.2+cpu torchaudio==2.2.2+cpu \
-    https://download.pytorch.org/whl/cpu/torchvision-0.17.2%2Bcpu-cp310-cp310-linux_x86_64.whl
-RUN pip install TTS==0.22.0 transformers flask sentencepiece numpy==1.22.0 scipy librosa==0.10.1 numba==0.57.1
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
 
+# Expose the port used by Flask
 EXPOSE 7860
+
+# Start the Flask app
 CMD ["python", "app.py"]
+