from datasets import load_dataset dataset = load_dataset('Seetha/visual_cs', split='train')