| from nvidia/cuda:12.1.1-devel-ubuntu22.04 | |
| run apt-get update && apt-get install -y \ | |
| python3.11 \ | |
| python3-pip \ | |
| ffmpeg \ | |
| libsndfile1 \ | |
| git \ | |
| && rm -rf /var/lib/apt/lists/* | |
| workdir /app | |
| copy requirements.txt . | |
| run pip install --no-cache-dir torch torchaudio --index-url https://download.pytorch.org/whl/cu121 | |
| run pip install --no-cache-dir -r requirements.txt | |
| copy . . | |
| expose 7860 | |
| cmd ["gunicorn", "--bind", "0.0.0.0:7860", "app:app", "--workers", "1", "--threads", "4", "--timeout", "600"] |