flow_inference.create_trocr_dataset

Provide a PyTorch dataset wrapper for TrOCR inference records.

Classes

TrOCRInferenceDataset(records, image_handler)

Represent Hugging Face records as a PyTorch dataset for TrOCR inference.