Skip to content
Snippets Groups Projects
Verified Commit 52a152ac authored by Yoann Schneider's avatar Yoann Schneider :tennis:
Browse files

Fix HTR inference

parent 4d50e8c6
No related branches found
No related tags found
1 merge request!402Fix HTR inference
Hide whitespace changes
Inline Side-by-side
dan/ocr/predict/inference.py
+ 1
− 2
View file @ 52a152ac
......@@ -365,8 +365,7 @@ def process_batch(
line_separators = parse_delimiters(line_separators)
# NER Entities separators
ner_separators = parse_tokens_pattern(tokens.values())
ner_separators = parse_tokens_pattern(tokens.values()) if tokens else None
# Predict
logger.info("Predicting...")
prediction = dan_model.predict(
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment