mirrors/docling
1
0
Fork 0
mirror of https://github.com/DS4SD/docling.git synced 2025-12-08 20:58:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: handle empty result from RapidOCR to avoid crash (#2264)

Browse Source 
Signed-off-by: Junehyuk Park <yuie@evonit.net>
This commit is contained in:
Yuie.
2025-09-15 17:04:33 +09:00
committed by GitHub
parent 10bb0aee2d
commit 609d902eef
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docling/models/rapid_ocr_model.py
View File

@@ -120,6 +120,9 @@ class RapidOcrModel(BaseOcrModel):
use_cls=self.options.use_cls, use_cls=self.options.use_cls,
use_rec=self.options.use_rec, use_rec=self.options.use_rec,
) )
if result is None or result.boxes is None:
_log.warning("RapidOCR returned empty result!")
continue
result = list( result = list(
zip(result.boxes.tolist(), result.txts, result.scores) zip(result.boxes.tolist(), result.txts, result.scores)
) )