temp

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 from marker.convert import convert_single_pdf
 from marker.models import load_all_models
 import surya.detection as detection
+import surya.layout as layout
 
 model_lst = load_all_models()
 
@@ -23,6 +24,30 @@ def batch_text_detection(images, model, processor, batch_size=None):
 detection.batch_text_detection = batch_text_detection
 
 
+def batch_layout_detection(
+    images, model, processor, detection_results=None, batch_size=None
+):
+    preds, orig_sizes = layout.batch_detection(
+        images, model, processor, batch_size=batch_size
+    )
+    id2label = model.config.id2label
+
+    results = []
+    for i in range(len(images)):
+        result = layout.parallel_get_regions(
+            preds[i],
+            orig_sizes[i],
+            id2label,
+            detection_results[i] if detection_results else None,
+        )
+        results.append(result)
+
+    return results
+
+
+layout.batch_layout_detection = batch_layout_detection
+
+
 @spaces.GPU
 def convert(file_path):
     print(file_path)