Harry Pham commited on
Commit
fd8d360
·
1 Parent(s): 9d7abf9

update OCR

Browse files
Files changed (1) hide show
  1. src/inference.py +16 -0
src/inference.py CHANGED
@@ -12,6 +12,22 @@ import numpy as np
12
  from pathlib import Path
13
  from ultralytics import RTDETR
14
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
15
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
16
  print(f"[INFO] Device: {DEVICE}")
17
 
 
12
  from pathlib import Path
13
  from ultralytics import RTDETR
14
 
15
+ import os
16
+ import logging
17
+ import cv2
18
+ import numpy as np
19
+ from paddleocr import PaddleOCR
20
+
21
+ # 1. Tắt check network của Paddle (Giúp app khởi động nhanh hơn trên Hugging Face)
22
+ os.environ["PADDLE_PDX_DISABLE_MODEL_SOURCE_CHECK"] = "True"
23
+
24
+ # 2. Tắt log rác của PaddleOCR (Thay thế cho tham số show_log=False đã bị xóa)
25
+ logging.getLogger("ppocr").setLevel(logging.ERROR)
26
+
27
+ print("[INFO] Loading PaddleOCR...")
28
+ # 3. Khởi tạo PaddleOCR (ĐÃ XÓA bỏ show_log=False)
29
+ _ocr_engine = PaddleOCR(use_angle_cls=True, lang='vi')
30
+
31
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
32
  print(f"[INFO] Device: {DEVICE}")
33