diff --git a/flash/vision/embedding/image_embedder_model.py b/flash/vision/embedding/image_embedder_model.py index 7a504a2fc58..b215065afe6 100644 --- a/flash/vision/embedding/image_embedder_model.py +++ b/flash/vision/embedding/image_embedder_model.py @@ -29,6 +29,7 @@ from flash.vision.embedding.model_map import _load_bolts_model, _models + class ImageEmbedderDataPipeline(TaskDataPipeline): """ >>> from flash.vision.embedding import ImageEmbedderDataPipeline