VietKien commited on
Commit
60d6bcc
·
verified ·
1 Parent(s): 83cd348

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -6,16 +6,21 @@ import cv2
6
  import numpy as np
7
  from PIL import Image
8
 
9
- # Thử import PaddleOCR, nếu chưa cài thì báo lỗi nhẹ
 
 
10
  try:
11
  from paddleocr import PaddleOCR
12
- # Khởi tạo model OCR (tải model lần đầu sẽ hơi lâu)
13
- # lang='vi' để hỗ trợ tiếng Việt
14
- ocr_engine = PaddleOCR(use_angle_cls=True, lang='vi', show_log=False)
15
  HAS_OCR = True
16
  except ImportError:
17
  HAS_OCR = False
18
  print("⚠️ CHƯA CÀI PADDLEOCR: Vui lòng chạy 'pip install paddlepaddle paddleocr'")
 
 
 
19
 
20
  # --- 1. CẤU HÌNH NGÔN NGỮ (TỪ ĐIỂN) ---
21
  TRANS = {
 
6
  import numpy as np
7
  from PIL import Image
8
 
9
+ # Thêm dòng này lên đầu file hoặc ngay dưới các lệnh import để tắt check mạng
10
+ os.environ["PADDLE_PDX_DISABLE_MODEL_SOURCE_CHECK"] = "True"
11
+
12
  try:
13
  from paddleocr import PaddleOCR
14
+ # Sửa: Bỏ 'show_log', đổi 'use_angle_cls' thành 'use_textline_orientation'
15
+ # Nếu vẫn báo lỗi tham số, chỉ cần dùng: PaddleOCR(lang='vi')
16
+ ocr_engine = PaddleOCR(use_textline_orientation=True, lang='vi')
17
  HAS_OCR = True
18
  except ImportError:
19
  HAS_OCR = False
20
  print("⚠️ CHƯA CÀI PADDLEOCR: Vui lòng chạy 'pip install paddlepaddle paddleocr'")
21
+ except Exception as e:
22
+ HAS_OCR = False
23
+ print(f"⚠️ Lỗi khởi tạo OCR: {e}")
24
 
25
  # --- 1. CẤU HÌNH NGÔN NGỮ (TỪ ĐIỂN) ---
26
  TRANS = {