From 1149d3ae08399e2c569404b862cc9d5b9f60d99b Mon Sep 17 00:00:00 2001
From: Nikos Livathinos <nli@zurich.ibm.com>
Date: Mon, 9 Dec 2024 11:12:28 +0100
Subject: [PATCH] fix: TableStructureModel: Refactor the artifacts path to use
 the new structure for fast/accurate model

Signed-off-by: Nikos Livathinos <nli@zurich.ibm.com>

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
---
 docling/models/table_structure_model.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/docling/models/table_structure_model.py b/docling/models/table_structure_model.py
index 23e5b4c3..e5ab1fe2 100644
--- a/docling/models/table_structure_model.py
+++ b/docling/models/table_structure_model.py
@@ -35,7 +35,9 @@ class TableStructureModel(BasePageModel):
         self.enabled = enabled
         if self.enabled:
             if self.mode == TableFormerMode.ACCURATE:
-                artifacts_path = artifacts_path / "fat"
+                artifacts_path = artifacts_path / "accurate"
+            else:
+                artifacts_path = artifacts_path / "fast"
 
             # Third Party
             import docling_ibm_models.tableformer.common as c