From 3c058c96a8118701e5649de5ceee4dca610f73de Mon Sep 17 00:00:00 2001 From: Pang Jin Hui Date: Mon, 21 Oct 2024 22:08:41 +0800 Subject: [PATCH] fix(demo layout model path) Signed-off-by: Pang Jin Hui --- demo/demo_layout_predictor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/demo_layout_predictor.py b/demo/demo_layout_predictor.py index 38ea40d..506125b 100644 --- a/demo/demo_layout_predictor.py +++ b/demo/demo_layout_predictor.py @@ -97,7 +97,7 @@ def main(args): # Download models from HF download_path = snapshot_download(repo_id="ds4sd/docling-models") - artifact_path = os.path.join(download_path, "model_artifacts/layout/beehive_v0.0.5") + artifact_path = os.path.join(download_path, "model_artifacts/layout/beehive_v0.0.5_pt") # Test the LayoutPredictor demo(logger, artifact_path, num_threads, img_dir, viz_dir)