update layout models

VikParuchuri · Nov 6, 2024 · 50f37dd · 50f37dd
1 parent cde7119
commit 50f37dd
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tabled/inference/models.py b/tabled/inference/models.py
@@ -1,4 +1,5 @@
 from surya.model.detection.model import load_model as load_det_model, load_processor as load_det_processor
+from surya.model.layout.model import load_model as load_layout_model, load_processor as load_layout_processor
 from surya.model.recognition.model import load_model as load_rec_model
 from surya.model.recognition.processor import load_processor as load_rec_processor
 from surya.model.table_rec.model import load_model as load_table_rec_model
@@ -9,8 +10,8 @@
 def load_detection_models():
     det_model = load_det_model()
     det_processor = load_det_processor()
-    layout_model = load_det_model(checkpoint=surya_settings.LAYOUT_MODEL_CHECKPOINT)
-    layout_processor = load_det_processor(checkpoint=surya_settings.LAYOUT_MODEL_CHECKPOINT)
+    layout_model = load_layout_model()
+    layout_processor = load_layout_processor()
     return det_model, det_processor, layout_model, layout_processor