Skip to content

Commit 3e0669b

Browse files
authored
Merge pull request #152 from mmacvicar/bugfix/ocr-no-detections
Fix OCR exception when no text detected
2 parents c985bcd + ef8aa1b commit 3e0669b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

surya/recognition.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@ def batch_recognition(images: List, languages: List[List[str]], model, processor
2626
assert all([isinstance(image, Image.Image) for image in images])
2727
assert len(images) == len(languages)
2828

29+
if len(images) == 0:
30+
return [], []
31+
2932
for l in languages:
3033
assert len(l) <= settings.RECOGNITION_MAX_LANGS, f"OCR only supports up to {settings.RECOGNITION_MAX_LANGS} languages per image, you passed {l}."
3134

0 commit comments

Comments
 (0)