diff --git a/deploy/pdserving/rec_local_server.py b/deploy/pdserving/rec_local_server.py index 58df5b32f831c6549c3e898d77fba6a31fc8eb94..90f4a0e5f9af2ed7deedb9b5afd4fadbcedf8df4 100644 --- a/deploy/pdserving/rec_local_server.py +++ b/deploy/pdserving/rec_local_server.py @@ -163,7 +163,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(