diff --git a/pipeline/persist.py b/pipeline/persist.py index c9904c3..511a2a6 100644 --- a/pipeline/persist.py +++ b/pipeline/persist.py @@ -9,9 +9,10 @@ DEFAULT_OUT = Path("output/v2") def save_result(result: dict, out_dir: Path | str = DEFAULT_OUT) -> Path: out_dir = Path(out_dir) out_dir.mkdir(parents=True, exist_ok=True) + from .ocr_qwen import MODEL_PATH as OCR_MODEL_ID result["_meta"] = { - "pipeline_version": "v1", - "ocr_model": "zai-org/GLM-OCR", + "pipeline_version": "v2", + "ocr_model": OCR_MODEL_ID, "generated_at": datetime.now(timezone.utc).isoformat(timespec="seconds"), } out_path = out_dir / f"{result['fichier']}.json"