chore: Replace the layout_predictor.predict_batch() with layout_predictor.predict() in a loop

Signed-off-by: Nikos Livathinos <nli@zurich.ibm.com>