* Save the batch-predict package in the image; create a separate Dockerfile for gpu * remove commented code