From 14e7ee6ed99b60ca369ef678afbe628fa520b275 Mon Sep 17 00:00:00 2001
From: Vik Paruchuri <vik.paruchuri@gmail.com>
Date: Fri, 15 Aug 2025 11:18:31 -0400
Subject: [PATCH] Avoid truncating layout and table

---
 surya/foundation/__init__.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/surya/foundation/__init__.py b/surya/foundation/__init__.py
index 9fa0c50..60ba803 100644
--- a/surya/foundation/__init__.py
+++ b/surya/foundation/__init__.py
@@ -819,6 +819,11 @@ class FoundationPredictor(BasePredictor):
                             ] or (
                                 drop_repeated_tokens
                                 and detect_repeat_token(predicted_tokens[p_idx])
+                                and task_names[p_idx]
+                                in [
+                                    TaskNames.ocr_with_boxes,
+                                    TaskNames.ocr_without_boxes,
+                                ]
                             )
                             if (
                                 token