| FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-devel as base | |
| RUN apt-get update && apt-get -y install git | |
| RUN apt-key adv --keyserver keyserver.ubuntu.com --recv-keys A4B469963BF863CC | |
| ENV HOME=/exp/fromage | |
| WORKDIR /exp/fromage | |
| COPY ./requirements.txt ./requirements.txt | |
| RUN python -m pip install -r ./requirements.txt | |
| RUN python -m pip install gradio | |
| COPY . . | |
| RUN chmod -R a+rwX . | |
| CMD ["uvicorn", "app:main", "--host", "0.0.0.0", "--port", "7860"] |