I tried using predict-on, but it is terrible