浏览代码

fix typos about parser.

Signed-off-by: zhanluxianshen <zhanluxianshen@163.com>
zhanluxianshen 2 月之前
父节点
当前提交
0bc31889cb
共有 2 个文件被更改,包括 1 次插入6 次删除
  1. 0 5
      dots_ocr/model/inference.py
  2. 1 1
      dots_ocr/parser.py

+ 0 - 5
dots_ocr/model/inference.py

@@ -1,8 +1,3 @@
-import json
-import io
-import base64
-import math
-from PIL import Image
 import requests
 from dots_ocr.utils.image_utils import PILimage_to_base64
 from openai import OpenAI

+ 1 - 1
dots_ocr/parser.py

@@ -153,7 +153,7 @@ class DotsOCRParser:
             min_pixels = min_pixels or MIN_PIXELS  # preprocess image to the final input
             max_pixels = max_pixels or MAX_PIXELS
         if min_pixels is not None: assert min_pixels >= MIN_PIXELS, f"min_pixels should >= {MIN_PIXELS}"
-        if max_pixels is not None: assert max_pixels <= MAX_PIXELS, f"max_pixels should <+ {MAX_PIXELS}"
+        if max_pixels is not None: assert max_pixels <= MAX_PIXELS, f"max_pixels should <= {MAX_PIXELS}"
 
         if source == 'image' and fitz_preprocess:
             image = get_image_by_fitz_doc(origin_image, target_dpi=self.dpi)