修正模型地址

2024-09-26 16:56:48 +08:00
parent 603b027ca6
commit b5aba0418b
2 changed files with 3 additions and 4 deletions
--- a/services/paddle_services/doc_dewarp/init.py
+++ b/services/paddle_services/doc_dewarp/init.py
@@ -1,7 +1,7 @@
-import os.path
+import os

 from onnxruntime import InferenceSession

-MODEL_PATH = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))),
+MODEL_PATH = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))),
                          'model', 'dewarp_model', 'doc_tr_pp.onnx')
 DOC_TR = InferenceSession(MODEL_PATH, providers=['CUDAExecutionProvider'], provider_options=[{'device_id': 0}])