Update Dockerfile to use Python base image and add input args

Update Dockerfile to use Python base image and add input args #2228