Add predictor for MultiBERT model, improve the trainer for the same model and expose the predicted score in the model's output