diff --git a/main.py b/main.py index 8e77298..d1b5987 100644 --- a/main.py +++ b/main.py @@ -38,7 +38,7 @@ gpu_available = paddle.device.is_compiled_with_cuda() print("GPU available:", gpu_available) -ocr = PaddleOCR(use_angle_cls=True, lang="ch", use_gpu=True) +ocr = PaddleOCR(use_angle_cls=True, lang="ch", use_gpu=conf.GPU_ENABLE) app = FastAPI() ALLOWED_FILE_TYPES = {"application/pdf", "image/jpeg", "image/png", "image/jpg"}