from datasets import load_dataset dataset = load_dataset("go_emotions", split="train") emotions = dataset.info.features['labels'].feature.names def query_emotion(start, end): rows = dataset[start:end] texts, labels, id = [rows[k] for k in rows.keys()] observations = [] for i, text in enumerate(texts): observations.append({ "text": text, "emotion": emotions[labels[i]], }) return observations