Ver código fonte

Merge pull request #146 from zhanluxianshen/fix-typos

fix typos about parser.
Qing Yan 2 meses atrás
pai
commit
c383de8e70
2 arquivos alterados com 1 adições e 6 exclusões
  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
 import requests
 from dots_ocr.utils.image_utils import PILimage_to_base64
 from dots_ocr.utils.image_utils import PILimage_to_base64
 from openai import OpenAI
 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
             min_pixels = min_pixels or MIN_PIXELS  # preprocess image to the final input
             max_pixels = max_pixels or MAX_PIXELS
             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 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:
         if source == 'image' and fitz_preprocess:
             image = get_image_by_fitz_doc(origin_image, target_dpi=self.dpi)
             image = get_image_by_fitz_doc(origin_image, target_dpi=self.dpi)