fix bug in predict were we weren't tf_transforming input