from datasets import load_dataset
 
# Load jsonl data from disk
dataset = load_dataset("json", data_files="train_dataset.json", split="train")