YiYiXu
/

florence-2-block

Model card Files Files and versions

YiYiXu HF Staff commited on Sep 26, 2025

Commit

778a168

·

verified ·

1 Parent(s): 3e5a8b3

Update block.py

Files changed (1) hide show

block.py +2 -2

block.py CHANGED Viewed

@@ -214,12 +214,12 @@ class Florence2ImageAnnotatorBlock(ModularPipelineBlocks):
             block_state.mask_image = None
         if block_state.annotation_output_type == "mask_overlay":
-            block_state.image = self.prepare_mask(
                 images, annotations, overlay=True, fill=fill
             )
         elif block_state.annotation_output_type == "bounding_box":
-            block_state.image = self.prepare_bounding_boxes(images, annotations)
         self.set_block_state(state, block_state)

             block_state.mask_image = None
         if block_state.annotation_output_type == "mask_overlay":
+            block_state.images = self.prepare_mask(
                 images, annotations, overlay=True, fill=fill
             )
         elif block_state.annotation_output_type == "bounding_box":
+            block_state.images = self.prepare_bounding_boxes(images, annotations)
         self.set_block_state(state, block_state)