wai572 commited on
Commit
8c9c5d9
·
1 Parent(s): a27cd33
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -65,7 +65,7 @@ def load_ocr_model():
65
  global reader
66
 
67
  # 使用する言語と、モデルの保存先ディレクトリを指定してReaderを初期化
68
- reader = PaddleOCR(lang="en", use_angle_cls=False, det=False)
69
  print(f"PaddleOCR model loaded successfully.")
70
 
71
 
@@ -210,7 +210,7 @@ async def analyze_image(image_paths: list[UploadFile]):
210
  text_mask = get_not_white_mask(img)
211
  masked_img = cv2.bitwise_and(img, img, mask=text_mask)
212
  # result = reader.readtext(candidate["img"])
213
- result = reader.ocr(masked_img)
214
  print(result)
215
  if len(result) > 0:
216
  ocr_results.append(result)
 
65
  global reader
66
 
67
  # 使用する言語と、モデルの保存先ディレクトリを指定してReaderを初期化
68
+ reader = PaddleOCR(lang="en", use_angle_cls=False)
69
  print(f"PaddleOCR model loaded successfully.")
70
 
71
 
 
210
  text_mask = get_not_white_mask(img)
211
  masked_img = cv2.bitwise_and(img, img, mask=text_mask)
212
  # result = reader.readtext(candidate["img"])
213
+ result = reader.ocr(masked_img, det=False)
214
  print(result)
215
  if len(result) > 0:
216
  ocr_results.append(result)