Update Dockerfile

Dockerfile

@@ -1,21 +1,36 @@
 FROM python:3.10-slim
 
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+
 WORKDIR /app
 
-# System deps
 RUN apt-get update && apt-get install -y \
     git \
     ffmpeg \
     libsndfile1 \
+    build-essential \
+    curl \
     && rm -rf /var/lib/apt/lists/*
 
 COPY requirements.txt .
 
-RUN pip install --upgrade pip
+RUN pip install --upgrade pip setuptools wheel
+
+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
 
+# Install MusicGen safely
+RUN pip install --no-cache-dir git+https://github.com/facebookresearch/audiocraft.git
+
 COPY . .
 
+RUN mkdir -p outputs
+
 EXPOSE 7860
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn","app:app","--host","0.0.0.0","--port","7860"]