# FROM python:3.10-slim FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime WORKDIR /app ENV DEBIAN_FRONTEND=noninteractive ENV TZ=Etc/UTC RUN apt-get update && apt-get install -y \ build-essential \ git \ python3-dev \ libsndfile1 \ && rm -rf /var/lib/apt/lists/* COPY requirements.txt . RUN pip install -r requirements.txt COPY inference.py . COPY modules ./modules COPY configs ./configs COPY hf_utils.py ./ COPY api.py ./ COPY runs ./runs COPY examples ./examples ENV PYTHONPATH=/app ENV HF_HUB_CACHE=/app/checkpoints/hf_cache ENV TORCH_HOME=/app/checkpoints ENV API_KEY=12345 EXPOSE 8000 CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "8000"]