Spaces:
Running
Running
Harry Pham commited on
Commit ·
fd8d360
1
Parent(s): 9d7abf9
update OCR
Browse files- 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 |
|