diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index 6cb5ce2..56d7f78 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -115,6 +115,8 @@ def get_ocr_layout(img_path): layout = [] ocr_result = OCR.ocr(img_path) ocr_result = ocr_result[0] + if not ocr_result: + return layout for segment in ocr_result: box = segment[0] box = _get_box(box)