fix: avoid potential divide by zero error

dynobo · Feb 9, 2025 · c7b17ed · c7b17ed
1 parent 2570c0c
commit c7b17ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/normcap/ocr/transformers/paragraph.py b/normcap/ocr/transformers/paragraph.py
@@ -14,8 +14,8 @@ def score(ocr_result: OcrResult) -> float:
     Returns:
         Score between 0-100 (100 = more likely).
     """
-    breaks = ocr_result.num_blocks + ocr_result.num_pars - 1
-    return 100 - (100 / (breaks))
+    breaks = max(1, ocr_result.num_blocks + ocr_result.num_pars - 1)
+    return 100 - (100 / breaks)
 
 
 def transform(ocr_result: OcrResult) -> str: