T5 MLM: Add next extra_id to end of label ids like it was done in training of T5 (see paper)

src/models/T5_MLM_label.py

@@ -15,7 +15,7 @@ def classify_entity(sentence, entity, labels):
 
     results = {}
     for label in labels:
-        label_ids = tokenizer(f"<extra_id_0> {label}", return_tensors="pt").input_ids
+        label_ids = tokenizer(f"<extra_id_0> {label} <extra_id_1>", return_tensors="pt").input_ids
         loss = model(input_ids=input_ids, labels=label_ids).loss.item()
         results[loss] = label