| # Set the platform flag if we're on ARM | |
| arch=$(uname -m) | |
| if [[ "$arch" == "arm64" || "$arch" == "aarch64" ]]; then | |
| platform_flag="--platform linux/amd64" | |
| else | |
| platform_flag="" | |
| fi | |
| # Build the Docker image | |
| docker build --progress=plain $platform_flag "$@" -t "openworm_ai" . | |