inference.py: add batch_size argument to rank()

inference.py: add batch_size argument to rank() #31