Update Dockerfile

Dockerfile

@@ -1,51 +1,45 @@
-# -------------------------------
-# Dockerfile for Voice & Chat AI
-# -------------------------------
-
-# Base image
+# ===== Base Image =====
 FROM python:3.10-slim
 
-# Set environment variables
+# ===== Environment Variables =====
 ENV LANG=C.UTF-8 \
     LC_ALL=C.UTF-8 \
     PYTHONUNBUFFERED=1 \
-    PIP_NO_CACHE_DIR=1
-
-# Install system dependencies
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential \
-    ffmpeg \
-    libsndfile1 \
-    git \
-    && rm -rf /var/lib/apt/lists/*
-
-# Upgrade pip
-RUN pip install --upgrade pip
-
-# Install Python dependencies
-RUN pip install \
-    numpy==1.25.2 \
-    torch==2.8.1 \
-    torchaudio==2.8.1 \
-    torchvision==0.15.2 \
-    transformers==4.38.0 \
-    gradio==3.55 \
-    sounddevice==0.4.6 \
-    soundfile==0.12.1 \
-    scipy==1.11 \
-    datasets==2.13 \
-    accelerate \
-    sentencepiece \
-    pybind11>=2.12
-
-# Set work directory
+    TORCH_CUDA_ARCH_LIST="6.1;7.5;8.6+PTX"
+
+# ===== System Dependencies =====
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        git \
+        wget \
+        ffmpeg \
+        libsndfile1 \
+        build-essential \
+        && rm -rf /var/lib/apt/lists/*
+
+# ===== Python Dependencies =====
+# Install specific compatible versions
+RUN pip install --no-cache-dir --upgrade pip setuptools wheel
+
+RUN pip install --no-cache-dir \
+    torch==2.2.0 \
+    torchaudio==2.2.0 \
+    torchvision==0.15.1 \
+    transformers==4.38 \
+    gradio>=3.41,<6 \
+    sounddevice>=0.4.6 \
+    soundfile>=0.12.1 \
+    scipy>=1.11 \
+    numpy<2 \
+    datasets>=2.13 \
+    accelerate
+
+# ===== Copy App Code =====
 WORKDIR /app
-
-# Copy your application files
 COPY . /app
 
-# Expose port for Gradio or API
+# ===== Expose Port (for Gradio if needed) =====
 EXPOSE 7860
 
-# Default command to run your app
+# ===== Entrypoint =====
 CMD ["python", "app.py"]