modosaic.text.postprocessor

modosaic.text.postprocessor

TextPostprocessor

Bases: ModalityPostprocessor[str]

Postprocessor that saves generated text captions.

modality property

modality

Return the text modality.

process

process(record, generated)

Create a text artifact for a generated caption.

Source code in modosaic/text/postprocessor.py

@override
def process(self, record: ImageRecord, generated: str) -> list[ExperimentArtifact[Any]]:
    """Create a text artifact for a generated caption."""
    logger.debug(f"Post-processing text for record sample {record.sample_id}")
    return [
        ExperimentArtifact.text(self.sample_path(record, ".txt"), generated)
    ]