Skip to content
Snippets Groups Projects
Commit 1392f95f authored by Yoann Schneider's avatar Yoann Schneider :tennis: Committed by Solene Tarride
Browse files

Support multiple predictions

parent bb413c0f
No related branches found
No related tags found
1 merge request!263Support multiple predictions
......@@ -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))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment