Update paddleocr.py (#10290)

解决多次识别pdf，受第一次页面最大数量影响的bug
PaddlePaddle · Jul 4, 2023 · b888862 · b888862
1 parent 1fa3575
commit b888862
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/paddleocr.py b/paddleocr.py
@@ -532,8 +532,9 @@ def ocr(self, img, det=True, rec=True, cls=True):
         # for infer pdf file
         if isinstance(img, list):
             if self.page_num > len(img) or self.page_num == 0:
-                self.page_num = len(img)
-            imgs = img[:self.page_num]
+                imgs=img
+            else:
+                imgs = img[:self.page_num]
         else:
             imgs = [img]
         if det and rec: