Dockerfile

FROM python:3.9

WORKDIR /code

RUN pip install aim

COPY stylegan_logs.tar.gz .
RUN tar -xzf stylegan_logs.tar.gz

RUN chmod -R 777 /code

CMD ["aim", "up", "--host", "0.0.0.0", "--port", "7860", "--workers", "2"]