Fixing code in batch predictions docs