chore: Change layout model path to a valid path (#43)

Signed-off-by: Pang Jin Hui <[email protected]>
DS4SD · Oct 29, 2024 · 0683efc · 0683efc
1 parent 09022d2
commit 0683efc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)