Update Dockerfile

Dockerfile

@@ -3,38 +3,36 @@ FROM python:3.11-slim
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 
-# 1. Install system dependencies
+# 1. Install system dependencies (root)
 RUN apt-get update && apt-get install -y \
     ffmpeg \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-# 2. Create the non-root user required by Hugging Face Spaces (UID 1000)
+# 2. Install Python dependencies globally (root).
+# This guarantees that Python can resolve all modules without relying on $PATH injection.
+COPY requirements.txt .
+RUN pip install --no-cache-dir torch torchaudio --index-url https://download.pytorch.org/whl/cpu
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 3. Create the unprivileged user required by Hugging Face Spaces (UID 1000)
 RUN useradd -m -u 1000 user
-USER user
 
-# Set environment variables for the new user
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH
+# 4. Pre-create the cache directory and assign ownership to the new user.
+# This ensures the user has write permissions when the _download function executes.
+RUN mkdir -p /home/user/.cache/whisper && chown -R user:user /home/user/.cache
 
+# 5. Switch context to the unprivileged user
+USER user
+ENV HOME=/home/user
 WORKDIR $HOME/app
 
-# 3. Copy requirements and transfer ownership to the new user
-COPY --chown=user requirements.txt .
-
-# 4. Install Python dependencies as the user
-RUN pip install --user --no-cache-dir torch torchaudio --index-url https://download.pytorch.org/whl/cpu
-RUN pip install --user --no-cache-dir -r requirements.txt
-
-# 5. Pre-cache Whisper WITHOUT loading it into PyTorch memory to avoid OOM limits.
-# Using whisper._download handles the checksum logic and places it in the user's cache directory.
-RUN python -c "import whisper, os; whisper._download(whisper._MODELS['medium'], os.path.expanduser('~/.cache/whisper'), False)"
+# 6. Safely pre-cache Whisper directly to disk without loading PyTorch tensors into memory
+RUN python -c "import whisper; whisper._download(whisper._MODELS['medium'], '/home/user/.cache/whisper', False)"
 
-# 6. Copy the rest of the application code
+# 7. Copy the application code and transfer ownership
 COPY --chown=user . .
 
-# Expose the correct Hugging Face port
 EXPOSE 7860
 
-# Start the Flask application
 CMD ["python", "backend.py"]