Dockerfile

FROM python:3.9

WORKDIR /code

RUN pip install aim

COPY mnist_logs.tar.gz .
RUN tar -xzf mnist_logs.tar.gz


RUN touch /.aim_profile 
RUN chmod -R 777 /code /.aim_profile
RUN aim telemetry off

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