Merge pull request #3 from icecraft/fix/paddle_not_ret

fix: paddle no ret
opendatalab · Jul 1, 2024 · 931657c · 931657c
2 parents 67528f4 + 3ec17e2
commit 931657c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/magic_doc/libs/version.py b/magic_doc/libs/version.py
@@ -1 +1 @@
-__version__ = "0.1.41"
+__version__ = "0.1.42"
diff --git a/magic_doc/model/parallel_paddle.py b/magic_doc/model/parallel_paddle.py
@@ -26,9 +26,8 @@ def _run_ocr_concurrently(self, chunks):
         def run_ocr(chunk, i):
             result = []
             for idx, img in chunk:
-                ocr_res = self.models[i](img)
-                if ocr_res:
-                    result.append((idx, ocr_res))
+                ocr_res = self.models[i](img) or []
+                result.append((idx, ocr_res))
             return result
 
         with ThreadPoolExecutor(max_workers=len(chunks)) as executor:

diff --git a/magic_doc/model/seq_paddle.py b/magic_doc/model/seq_paddle.py
@@ -15,7 +15,7 @@ def __call__(self, params):
         """
         results = [] 
         for idx, img in params:
-            ocr_res = self.model(img)
+            ocr_res = self.model(img) or []
             results.append((idx, ocr_res))
 
         return results