refactor: use org--name in artifacts-path (#912)

use org--name in artifacts-path Signed-off-by: Michele Dolfi <[email protected]>
DS4SD · Feb 7, 2025 · 02faf53 · 02faf53
1 parent 90b766e
commit 02faf53
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/docling/models/code_formula_model.py b/docling/models/code_formula_model.py
@@ -62,7 +62,7 @@ class CodeFormulaModel(BaseItemAndImageEnrichmentModel):
         Processes the given batch of elements and enriches them with predictions.
     """
 
-    _model_repo_folder = "CodeFormula"
+    _model_repo_folder = "ds4sd--CodeFormula"
     elements_batch_size = 5
     images_scale = 1.66  # = 120 dpi, aligned with training data resolution
     expansion_factor = 0.03

diff --git a/docling/models/document_picture_classifier.py b/docling/models/document_picture_classifier.py
@@ -56,7 +56,7 @@ class DocumentPictureClassifier(BaseEnrichmentModel):
         Processes a batch of elements and adds classification annotations.
     """
 
-    _model_repo_folder = "DocumentFigureClassifier"
+    _model_repo_folder = "ds4sd--DocumentFigureClassifier"
     images_scale = 2
 
     def __init__(

diff --git a/docling/models/layout_model.py b/docling/models/layout_model.py
@@ -22,7 +22,7 @@
 
 
 class LayoutModel(BasePageModel):
-    _model_repo_folder = "docling-models"
+    _model_repo_folder = "ds4sd--docling-models"
     _model_path = "model_artifacts/layout"
 
     TEXT_ELEM_LABELS = [

diff --git a/docling/models/table_structure_model.py b/docling/models/table_structure_model.py
@@ -23,7 +23,7 @@
 
 
 class TableStructureModel(BasePageModel):
-    _model_repo_folder = "docling-models"
+    _model_repo_folder = "ds4sd--docling-models"
     _model_path = "model_artifacts/tableformer"
 
     def __init__(