diff --git a/deploy/pdserving/ocr_local_server.py b/deploy/pdserving/ocr_local_server.py index a4fb540b1e7bb6426b922367f3ddf7a239adf31f..b506790f2a3b7d6b4955bfc27640892f9104b24f 100644 --- a/deploy/pdserving/ocr_local_server.py +++ b/deploy/pdserving/ocr_local_server.py @@ -110,7 +110,7 @@ class OCRService(WebService): if __name__ == "__main__": ocr_service = OCRService(name="ocr") - ocr_service.load_model_config("ocr_rec_model") + ocr_service.load_model_config(global_args.rec_model_dir) ocr_service.init_rec() if global_args.use_gpu: ocr_service.prepare_server(