I am writing a program in Python. What it does is to train a question-answering model, and it uses a custom-made dataset from a JSON file that I created. However, I ran into the error from the full traceback below:

Traceback (most recent call last):
  File "C:UserschenpDocumentsMLmachineLearning.py", line 52, in <module>
    trainer.train()
  File "C:UserschenpAppDataLocalProgramsPythonPython310libsite-packagestransformerstrainer.py", line 1859, in train
    return inner_training_loop(
  File "C:UserschenpAppDataLocalProgramsPythonPython310libsite-packagestransformerstrainer.py", line 2203, in _inner_training_loop
    tr_loss_step = self.training_step(model, inputs)
  File "C:UserschenpAppDataLocalProgramsPythonPython310libsite-packagestransformerstrainer.py", line 3138, in training_step
    loss = self.compute_loss(model, inputs)
  File "C:UserschenpAppDataLocalProgramsPythonPython310libsite-packagestransformerstrainer.py", line 3179, in compute_loss
    raise ValueError(
ValueError: The model did not return a loss from the inputs, only the following keys: start_logits,end_logits. For reference, the inputs it received are input_ids,attention_mask.
  0%|          | 0/9 [00:00<?, ?it/s]

from the offending line:

trainer.train()

And this is my full code:

# https://www.mlexpert.io/blog/alpaca-fine-tuning
# https://wellsr.com/python/fine-tuning-huggingface-models-in-tensorflow-keras/
# https://learnopencv.com/fine-tuning-bert/
# https://medium.com/@karary/nlp-fine-tune-question-answering-model-%E5%AF%A6%E4%BD%9C-3-model-training-%E5%84%B2%E5%AD%98%E8%88%87-inference-13d2a5bf5c32
# https://medium.com/@anyuanay/fine-tuning-the-pre-trained-bert-model-in-hugging-face-for-question-answering-8edc76890ce0
import transformers as tf
import datasets as ds
import pandas as pd
import numpy as np
import torch
import json
 
############## Check if CUDA is enabled. ################
hasCUDA=torch.cuda.is_available()
print(f"CUDA Enabled? {hasCUDA}")
device="cuda" if hasCUDA else "cpu"      
 
############## Loading file and populating data ################
fileName="qna.json"
sampleDS2=ds.load_dataset("json", data_files=fileName, split="train")


############## Model ##########################################
modelName="distilbert/distilbert-base-cased"     #or replace the model name with whatever you feel like.
# config=tf.AutoConfig.from_pretrained(modelName+"/config.json")
model=tf.AutoModelForQuestionAnswering.from_pretrained(modelName)
tokenizer=tf.AutoTokenizer.from_pretrained(modelName)

############## Encoding and Tokenizing #######################################
sampleDS2=sampleDS2.map(lambda batch: tokenizer(sampleDS2["question"], truncation=True, padding='max_length', max_length=512), batched=True)
sampleDS2.set_format("torch", columns=["input_ids", "attention_mask", "question"])
# sampleDS2.rename_column("answer", "labels")
print(sampleDS2)
############## Training #######################################
trnArgs=tf.TrainingArguments(
    output_dir="./",
    evaluation_strategy="epoch",
    save_strategy="epoch",
    learning_rate=2e-5,
    num_train_epochs=3,
    remove_unused_columns=True,
    fp16=True
)
 
trainer=tf.Trainer(
    model=model,
    args=trnArgs,
    train_dataset=sampleDS2, 
    eval_dataset=sampleDS2,
    tokenizer=tokenizer
)
trainer.train()

Here is a sample of the JSON file that I made:

{"question": "Who wrote Charlie and the Chocolate Factory?", "labels": "Roald Dahl"}
{"question": "Name a few ways to treat constipation naturally.", "labels": "Exercise regularly, eat more fibers, and drink more water."}
{"question": "Where is the longest roller coaster located?", "labels": "Nagashima, Japan. The name of the coaster is Steel Dragon 2000."}
{"question": "Who murdered JFK?", "labels": "It is said to be Harvey Oswald."}
{"question": "What are the 11 herbs and spices that Colonel Sanders used in KFC?", "labels": "Nobody knows, as it's a secret."}
{"question": "Who wrote Les Miserables?", "labels": "Victor Hugo"}
{"question": "What is the Watergate Scandal?", "labels": "The Watergate scandal was a significant political controversy in the United States during the presidency of Richard Nixon from 1972 to 1974, ultimately resulting in Nixon's resignation. It originated from attempts by the Nixon administration to conceal its involvement in the June 17, 1972, break-in at the Democratic National Committee headquarters located in the Watergate Office Building in Washington, D.C."}
{"question": "What is Obama's most famous quote?", "labels": "'Yes we can!'"}
{"question": "Where did the 2008 Olympic take place?", "labels": "Beijing"}
{"question": "Lentils and Chickpeas are what kind of food?", "labels": "Beans"}
{"question": "Who was the disciple that Jesus loved?", "labels": "John"}
{"question": "Why did the Boston Tea Party happen?", "labels": "The colonists were unhappy with the tax and restrictions imposed by the British colonists."}
{"question": "What was the effect of the Boston Tea Party?", "labels": "The British imposed a new Intolerable Act, and tensions between the colonies and the British escalated."}
{"question": "Who conquered the Aztec Empire?", "labels": "Hernan Cortes"}
{"question": "What is the longest flight as of 2024?", "labels": "Singapore to New York, operated by Singapore Airlines."}
{"question": "Name a few infamous Roman dictators.", "labels": "Caligula, Nero, and Tiberius."}
{"question": "Where did the early Hungarians come from?", "labels": "They originated from the Uralic region as nomads, and then migrated to Central Europe's Carpathian basin."}
{"question": "Where is the fastest roller coaster located?", "labels": "Abu Dhabi, and the coaster is known as F1."}
{"question": "What are some popular painting in Uffizi Gallery?", "labels": "The Birth of Venus, Madonna of the Goldfinch, and Judith and Holofernes."}
{"question": "Who wrote A Christmas Carol and Oliver Twist?", "labels": "Charles Dickens"}

Any suggestions for this fix is welcome.