runtime error

Exit code: 1. Reason: Generating train split: 0 examples [00:00, ? examples/s] Generating train split: 1 examples [00:00, 154.17 examples/s] Some weights of BertForSequenceClassification were not initialized from the model checkpoint at google-bert/bert-base-cased and are newly initialized: ['classifier.bias', 'classifier.weight'] You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference. Map: 0%| | 0/1 [00:00<?, ? examples/s] Map: 0%| | 0/1 [00:00<?, ? examples/s] Traceback (most recent call last): File "/home/user/app/app.py", line 20, in <module> tokenized_datasets = dataset.map(tokenize_function, batched=True) File "/usr/local/lib/python3.10/site-packages/datasets/dataset_dict.py", line 866, in map { File "/usr/local/lib/python3.10/site-packages/datasets/dataset_dict.py", line 867, in <dictcomp> k: dataset.map( File "/usr/local/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 560, in wrapper out: Union["Dataset", "DatasetDict"] = func(self, *args, **kwargs) File "/usr/local/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 3035, in map for rank, done, content in Dataset._map_single(**dataset_kwargs): File "/usr/local/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 3438, in _map_single batch = apply_function_on_filtered_inputs( File "/usr/local/lib/python3.10/site-packages/datasets/arrow_dataset.py", line 3300, in apply_function_on_filtered_inputs processed_inputs = function(*fn_args, *additional_args, **fn_kwargs) File "/home/user/app/app.py", line 17, in tokenize_function return tokenizer(examples["prompt"], padding="max_length", truncation=True) File "/usr/local/lib/python3.10/site-packages/datasets/formatting/formatting.py", line 277, in __getitem__ value = self.data[key] KeyError: 'prompt'

Container logs: