Dockerfile

FROM python:3.9

WORKDIR /code

RUN pip install aim

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

RUN touch /.aim_profile 
RUN chown 1000:1000 /code /.aim_profile
RUN aim telemetry off

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