Dockerfile

FROM ghcr.io/replicate/cog:0.5.0

COPY . /src
WORKDIR /src

RUN pip install -r requirements.txt

ENTRYPOINT ["python", "predict.py"]