Update Dockerfile

Dockerfile

@@ -1,20 +1,34 @@
-# Stage 1: Build
-FROM python:3.9-slim as builder
+# Use an official Python runtime as a parent image
+FROM python:3.9-slim
+
+# Set environment variables for Hugging Face cache
+ENV TRANSFORMERS_CACHE=/app/.cache
+ENV HF_DATASETS_CACHE=/app/.cache
 
+# Create the cache directory
+RUN mkdir -p /app/.cache
+
+# Set the working directory in the container
 WORKDIR /app
+
+# Copy the requirements file into the container
 COPY requirements.txt .
-RUN pip install --no-cache-dir --user -r requirements.txt
 
-# Stage 2: Final Image
-FROM python:3.9-slim
-WORKDIR /app
+# Install system dependencies (required for soundfile and other libraries)
+RUN apt-get update && apt-get install -y \
+    libsndfile1 \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy only the necessary files from the builder stage
-COPY --from=builder /root/.local /root/.local
+# Copy the current directory contents into the container
 COPY . .
 
-# Ensure scripts in .local are usable
-ENV PATH=/root/.local/bin:$PATH
+# Pre-download the Whisper model during the build process
+RUN python -c "from transformers import WhisperProcessor, WhisperForConditionalGeneration; \
+    WhisperProcessor.from_pretrained('openai/whisper-large-v3', cache_dir='/app/.cache'); \
+    WhisperForConditionalGeneration.from_pretrained('openai/whisper-large-v3', cache_dir='/app/.cache')"
 
 # Expose port 7860 (default port for Gradio)
 EXPOSE 7860