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

Signed-off-by: Pang Jin Hui <[email protected]>

Author: arceushui

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)