Put train or test in html file.