Skip to content
Snippets Groups Projects

Support multiple predictions

Merged Yoann Schneider requested to merge correctly-parse-confidences into main
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -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))
Loading