Fix entity splitting

dan/predict/prediction.py

@@ -326,7 +326,8 @@ def process_image(
                 "text": f"{text[current: next_token]}".replace("\n", " "),
                 "confidence_ner": f"{np.around(np.mean(char_confidences[current : next_token]), 2)}",
             }
-            for current, next_token in pairwise(index + [0])
+            # We go up to -1 so that the last token matches until the end of the text
+            for current, next_token in pairwise(index + [-1])
         ]
         result["confidences"]["total"] = np.around(np.mean(char_confidences), 2)