Update block.py

block.py

@@ -37,11 +37,13 @@ class Florence2ImageAnnotatorBlock(ModularPipelineBlocks):
                 type_hint=Union[Image.Image, List[Image.Image]],
                 required=True,
                 description="Image(s) to annotate",
+                metadata={"mellon":"image"},
             ),
             InputParam(
                 "annotation_task",
                 type_hint=Union[str, List[str]],
                 default="<REFERRING_EXPRESSION_SEGMENTATION>",
+                metadata={"mellon":"dropdown"},
                 description="""Annotation Task to perform on the image.
                 Supported Tasks:
 
@@ -63,6 +65,7 @@ class Florence2ImageAnnotatorBlock(ModularPipelineBlocks):
                 "annotation_prompt",
                 type_hint=Union[str, List[str]],
                 required=True,
+                metadata={"mellon":"textbox"},
                 description="""Annotation Prompt to provide more context to the task.
                 Can be used to detect or segment out specific elements in the image
                 """,
@@ -71,6 +74,7 @@ class Florence2ImageAnnotatorBlock(ModularPipelineBlocks):
                 "annotation_output_type",
                 type_hint=str,
                 default="mask_image",
+                metadata={"mellon":"dropbox"},
                 description="""Output type from annotation predictions. Availabe options are
                 annotation:
                     - raw annotation predictions from the model based on task type.
@@ -88,6 +92,7 @@ class Florence2ImageAnnotatorBlock(ModularPipelineBlocks):
                 required=True,
                 default=False,
                 description="",
+                metadata={"mellon":"checkbox"},
             ),
             InputParam(
                 "fill",
@@ -109,6 +114,7 @@ class Florence2ImageAnnotatorBlock(ModularPipelineBlocks):
                 "images",
                 type_hint=Image,
                 description="Annotated input Image(s)",
+                metadata={"mellon":"image"},
             ),
         ]