Upload processing_kosmos2.py

processing_kosmos2.py

@@ -544,23 +544,7 @@ def extract_entities_with_patch_indices(text):
                 entity = f"<patch_index_{bbox[0]}><patch_index_{bbox[1]}>"
                 entities_with_patch_indices.append((entity, span, [bbox]))
 
-    def cleanup_spaces(text, entities):
-        new_text = text.strip()
-
-        leading_spaces = text - text.lstrip(text)
-
-        new_entities = []
-        for entity_name, (start, end), bboxes in entities:
-
-            start = start - leading_spaces + (entity_name.lstrip(entity_name))
-            end = end - leading_spaces - (entity_name.rstrip(entity_name))
-            entity_name = entity_name.strip()
-
-            new_entities.append((entity_name, (start, end), bboxes))
-
-        return new_text, new_entities
-
-    return cleanup_spaces(entities_with_patch_indices)
+    return entities_with_patch_indices
 
 
 # TODO: Be careful