Dockerfile

FROM python:3.8-slim

WORKDIR /code

RUN pip install --upgrade pip

RUN pip install diffusers transformers

RUN pip install fastapi uvicorn

CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "80"]