FROM public.ecr.aws/c9e4w0g3/notebook-servers/jupyter-tensorflow:2.6.0-cpu-py38

COPY training.py /
ENTRYPOINT ["python", "/training.py"]