feat: 同步报销流程与工作台改动

server/scripts/paddle_ocr_worker.py

@@ -21,6 +21,7 @@ def parse_args() -> argparse.Namespace:
     parser.add_argument("--lang", default="ch")
     parser.add_argument("--text-detection-model", default="PP-OCRv5_mobile_det")
     parser.add_argument("--text-recognition-model", default="PP-OCRv5_mobile_rec")
+    parser.add_argument("--device", default=os.environ.get("OCR_DEVICE", ""))
     parser.add_argument("--enable-mkldnn", action="store_true")
     return parser.parse_args()
 
@@ -100,16 +101,20 @@ def build_document(input_path: str, results: list[Any]) -> dict[str, Any]:
 
 def main() -> int:
     args = parse_args()
-    ocr = PaddleOCR(
-        text_detection_model_name=args.text_detection_model,
-        text_recognition_model_name=args.text_recognition_model,
-        use_doc_orientation_classify=False,
-        use_doc_unwarping=False,
-        use_textline_orientation=False,
-        lang=args.lang,
+    ocr_options = {
+        "text_detection_model_name": args.text_detection_model,
+        "text_recognition_model_name": args.text_recognition_model,
+        "use_doc_orientation_classify": False,
+        "use_doc_unwarping": False,
+        "use_textline_orientation": False,
+        "lang": args.lang,
         # PaddlePaddle 3.3.x CPU oneDNN can fail on PP-OCRv5 static inference.
-        enable_mkldnn=args.enable_mkldnn,
-    )
+        "enable_mkldnn": args.enable_mkldnn,
+    }
+    configured_device = str(args.device or "").strip()
+    if configured_device:
+        ocr_options["device"] = configured_device
+    ocr = PaddleOCR(**ocr_options)
 
     documents = []
     for input_path in args.inputs: