added score calculation
Browse files
README.md
CHANGED
@@ -219,7 +219,7 @@ tokens_starred, text_starred = preprocess_text(
|
|
219 |
)
|
220 |
|
221 |
|
222 |
-
segments, blank_id = get_alignments(
|
223 |
emissions,
|
224 |
tokens_starred,
|
225 |
alignment_dictionary,
|
@@ -227,6 +227,6 @@ segments, blank_id = get_alignments(
|
|
227 |
|
228 |
spans = get_spans(tokens_starred, segments, alignment_tokenizer.decode(blank_id))
|
229 |
|
230 |
-
word_timestamps = postprocess_results(text_starred, spans, stride)
|
231 |
|
232 |
```
|
|
|
219 |
)
|
220 |
|
221 |
|
222 |
+
segments, scores, blank_id = get_alignments(
|
223 |
emissions,
|
224 |
tokens_starred,
|
225 |
alignment_dictionary,
|
|
|
227 |
|
228 |
spans = get_spans(tokens_starred, segments, alignment_tokenizer.decode(blank_id))
|
229 |
|
230 |
+
word_timestamps = postprocess_results(text_starred, spans, stride, scores)
|
231 |
|
232 |
```
|