Update Dockerfile

Dockerfile

@@ -1,59 +1,29 @@
-# ─────────────────────────────────────────────────────────────
-# Dockerfile.cpu.fastapi — CPU-only FastAPI deployment for Dia
-# Build: docker build . -f Dockerfile.cpu.fastapi -t dia-cpu-fastapi
-# Run:   docker run --rm -p 7860:7860 dia-cpu-fastapi
-# ─────────────────────────────────────────────────────────────
-
-# 1) Base image
 FROM python:3.10-slim
 
-# 2) Non-interactive apt
-ENV DEBIAN_FRONTEND=noninteractive
+# Set environment variables
+ENV HF_HOME=/app/hf
+ENV PYTHONUNBUFFERED=1
+ENV PIP_NO_CACHE_DIR=1
 
-# 3) System dependencies
-RUN apt-get update && \
-    apt-get install -y --no-install-recommends \
-      libsndfile1 \
-      ffmpeg \
-      curl \
-      git \
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    ffmpeg \
+    libsndfile1 \
+    git \
     && rm -rf /var/lib/apt/lists/*
 
-# 4) Set workdir
+# Set working directory
 WORKDIR /app
 
-# 5) Copy and install Python deps
-COPY requirements.txt ./
-RUN pip install --upgrade pip && \
-    # CPU-only PyTorch wheels
-    pip install torch torchaudio --index-url https://download.pytorch.org/whl/cpu && \
-    # The rest of your libs (including descript-audio-codec, fastapi, uvicorn, etc.)
-    pip install --no-cache-dir -r requirements.txt
-
-# 6) Pre-download DAC + Dia model weights into a shared cache
-RUN mkdir -p /app/.cache/huggingface /app/models/dac && \
-    chmod -R 777 /app/.cache /app/models/dac && \
-    # 6a) Download DAC codebook
-    python -m dac download && \
-    # 6b) Download Dia backbone
-    python - <<EOF
-from huggingface_hub import snapshot_download
-snapshot_download(repo_id="nari-labs/Dia-1.6B", cache_dir="/app/.cache/huggingface")
-EOF
+# Copy requirements and install Python packages
+COPY requirements.txt .
+RUN pip install --root-user-action=ignore -r requirements.txt
 
-# 7) Copy your FastAPI app
+# Copy app files
 COPY . .
 
-# 8) Env vars so Dia & HF hub pick up the right cache
-ENV HF_HOME=/app/.cache/huggingface
-ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
-ENV XDG_CACHE_HOME=/app/.cache/huggingface
-ENV DIA_DAC_PATH=/app/models/dac
-
-# 9) Expose Uvicorn port
+# Expose port
 EXPOSE 7860
 
-# 10) Launch FastAPI via Uvicorn
-WORKDIR /app
+# Run the app
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
-