ydshieh
/

kosmos-2-patch14-224

Image-Text-to-Text

Inference Endpoints

Model card Files Files and versions Community

ydshieh HF staff commited on Aug 19, 2023

Commit

04514b5

·

1 Parent(s): a2a10d5

Upload processing_kosmos2.py

Files changed (1) hide show

processing_kosmos2.py +1 -1

processing_kosmos2.py CHANGED Viewed

@@ -574,7 +574,7 @@ def clean_text_and_extract_entities_with_bboxes(text, num_patches_per_side=32):
         adjusted_entity = adjust_entity_positions(entity, text)
         bboxes_in_coords = list(map(lambda bbox: patch_index_to_coordinate(bbox[0], bbox[1], num_patches_per_side), bboxes))
-        entities.append((adjusted_entity) + (bboxes_in_coords,))
     def cleanup_spaces(text, entities):
         new_text = text.strip()

         adjusted_entity = adjust_entity_positions(entity, text)
         bboxes_in_coords = list(map(lambda bbox: patch_index_to_coordinate(bbox[0], bbox[1], num_patches_per_side), bboxes))
+        entities.append(adjusted_entity + (bboxes_in_coords,))
     def cleanup_spaces(text, entities):
         new_text = text.strip()