diff --git a/scripts/tools/backend_config.py b/scripts/tools/backend_config.py index e34c368..72e7901 100644 --- a/scripts/tools/backend_config.py +++ b/scripts/tools/backend_config.py @@ -131,7 +131,7 @@ def detect_gpu(timeout: float = 45.0) -> tuple[bool, str]: if not parse_bool(os.environ.get("CNCAPTCHA_STRICT_GPU_DETECT"), True): return True, output - model_name = os.environ.get("CNCAPTCHA_GPU_OCR_MODEL", "PP-OCRv5_server_rec") + model_name = os.environ.get("CNCAPTCHA_GPU_OCR_MODEL", "PP-OCRv6_tiny_rec") device = os.environ.get("CNCAPTCHA_GPU_OCR_DEVICE", "gpu:0") engine = os.environ.get("CNCAPTCHA_GPU_OCR_ENGINE", "paddle_dynamic") ocr_probe = ( @@ -206,9 +206,9 @@ def resolve_backend_config(source: str = "env") -> BackendConfig: minimum=1, ), cpu_model=os.environ.get("CNCAPTCHA_CPU_OCR_MODEL", "hybrid"), - cpu_fast_model=os.environ.get("CNCAPTCHA_CPU_OCR_FAST_MODEL", "PP-OCRv5_mobile_rec"), - cpu_fallback_model=os.environ.get("CNCAPTCHA_CPU_OCR_FALLBACK_MODEL", "PP-OCRv5_server_rec"), - gpu_model=os.environ.get("CNCAPTCHA_GPU_OCR_MODEL", "PP-OCRv5_server_rec"), + cpu_fast_model=os.environ.get("CNCAPTCHA_CPU_OCR_FAST_MODEL", "PP-OCRv6_tiny_rec"), + cpu_fallback_model=os.environ.get("CNCAPTCHA_CPU_OCR_FALLBACK_MODEL", "PP-OCRv6_medium_rec"), + gpu_model=os.environ.get("CNCAPTCHA_GPU_OCR_MODEL", "PP-OCRv6_tiny_rec"), gpu_device=os.environ.get("CNCAPTCHA_GPU_OCR_DEVICE", "gpu:0"), constrained_decode=parse_bool(os.environ.get("CNCAPTCHA_OCR_CONSTRAINED"), True), gpu_available=gpu_available, diff --git a/scripts/tools/ppocr_cpu_pool_worker.py b/scripts/tools/ppocr_cpu_pool_worker.py index d1a867a..70002dc 100644 --- a/scripts/tools/ppocr_cpu_pool_worker.py +++ b/scripts/tools/ppocr_cpu_pool_worker.py @@ -13,7 +13,7 @@ import numpy as np ROOT = Path(__file__).resolve().parents[2] -MODEL_NAME = os.environ.get("CNCAPTCHA_CPU_OCR_MODEL", "PP-OCRv5_server_rec") +MODEL_NAME = os.environ.get("CNCAPTCHA_CPU_OCR_MODEL", "PP-OCRv6_tiny_rec") ENGINE = os.environ.get("CNCAPTCHA_CPU_OCR_ENGINE", "paddle_dynamic") WORKERS = int(os.environ.get("CNCAPTCHA_CPU_OCR_WORKERS", "3")) CACHE_NAME = os.environ.get("CNCAPTCHA_CPU_PADDLEX_CACHE", ".paddlex_cache_cpu")