Train Bart for Conditional Generation (e.g. Summarization)

Hi

I tried to use above code but facing below error. Can someone advise

Below part of the code is raising error

dataset_ftrs = dataset.map(get_features, batched=True)

Error details:
Map: 0%
0/287113 [00:00<?, ? examples/s]

KeyError Traceback (most recent call last)
in <cell line: 1>()
----> 1 dataset_ftrs = dataset.map(get_features, batched=True)

8 frames
/usr/local/lib/python3.10/dist-packages/datasets/formatting/formatting.py in getitem(self, key)
268
269 def getitem(self, key):
→ 270 value = self.data[key]
271 if key in self.keys_to_format:
272 value = self.format(key)

KeyError: ‘text’