Right now, the user has to keep their tab open, to make sure that they can train and predict on a custom dataset of their own.