Removed the button, now tap to predict works too