From 1392f95f350aaba1f0aac61164a5eb7fcae55b61 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Tue, 29 Aug 2023 10:39:24 +0200
Subject: [PATCH] Support multiple predictions
---
dan/ocr/predict/prediction.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/dan/ocr/predict/prediction.py b/dan/ocr/predict/prediction.py
index 064eb749..91b4895f 100644
--- a/dan/ocr/predict/prediction.py
+++ b/dan/ocr/predict/prediction.py
@@ -339,7 +339,7 @@ def process_batch(
# Return mean confidence score
if confidence_score:
result["confidences"] = {}
- char_confidences = prediction["confidences"][0]
+ char_confidences = prediction["confidences"][idx]
text = result["text"]
start_tokens, end_tokens = zip(*list(tokens.values()))
end_tokens = list(filter(bool, end_tokens))
--
GitLab