$(aws ecr get-login --region $region --no-include-email)
# Copy Dockerfiles
!cp Dockerfile.* build/

# Copy training script and config file
!cp cifar_10_keras.py build/
!cp keras.json build/

# Building the image
!cd build docker build -t $image_tag -f $dockerfile . cd ..

# Assign a tag to the Docker image
!docker tag $image_tag $account.dkr.ecr.$region.amazonaws.com/$repo_name:latest

# Push docker image to ECR
!docker push $account.dkr.ecr.$region.amazonaws.com/$repo_name:latest