Upload 3 files

Dockerfile

@@ -0,0 +1,37 @@
+FROM python:3.10
+
+# System deps
+RUN apt-get update && apt-get install -y \
+    ffmpeg \
+    git \
+    libgl1 \
+    libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
+
+# User
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH=/home/user/.local/bin:$PATH
+WORKDIR /app
+
+# Upgrade pip
+RUN pip install --upgrade pip setuptools wheel
+
+# Install torch FIRST (CPU)
+RUN pip install \
+    torch==2.1.0+cpu \
+    torchaudio==2.1.0+cpu \
+    --index-url https://download.pytorch.org/whl/cpu
+
+# 🔥 INSTALL AUDIOCRAFT FROM GITHUB (CRITICAL)
+RUN pip install \
+    git+https://github.com/facebookresearch/audiocraft.git@v1.3.0
+
+# App deps
+COPY --chown=user requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy app
+COPY --chown=user . /app
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,44 @@
+from fastapi import FastAPI
+
+import audiocraft
+print("Audiocraft version:", audiocraft.__version__)
+
+from audiocraft.models import MusicGen
+from audiocraft.data.audio import audio_write
+import uuid
+import os
+
+app = FastAPI()
+
+MODEL_NAME = "musicgen-medium"
+OUTPUT_DIR = "outputs"
+os.makedirs(OUTPUT_DIR, exist_ok=True)
+
+print("Loading MusicGen model...")
+model = MusicGen.get_pretrained(MODEL_NAME)
+model.set_generation_params(
+    duration=30,   # seconds
+    temperature=1.0,
+    top_k=250,
+    top_p=0.0
+)
+
+@app.post("/generate")
+async def generate(prompt: str):
+    wav = model.generate([prompt])[0]
+
+    filename = f"{uuid.uuid4().hex}"
+    path = os.path.join(OUTPUT_DIR, filename)
+
+    audio_write(
+        path,
+        wav.cpu(),
+        model.sample_rate,
+        strategy="loudness",
+        loudness_compressor=True
+    )
+
+    return {
+        "prompt": prompt,
+        "file": f"{filename}.wav"
+    }

requirements.txt

@@ -0,0 +1,5 @@
+fastapi
+uvicorn
+numpy
+scipy
+soundfile