diff --git a/paddle_hub/data/type.py b/paddle_hub/data/type.py index c4d992539c38c44e8376ff91688d345da787e01c..020399815feaf0aa1038bc1db5f9862a28f80f7a 100644 --- a/paddle_hub/data/type.py +++ b/paddle_hub/data/type.py @@ -65,6 +65,8 @@ class DataType(Enum): if data_type == DataType.IMAGE: return ImageReader + elif data_type == DataType.TEXT: + return TextReader else: type_str = DataType.str(data_type) logger.critical( @@ -78,3 +80,9 @@ class ImageReader: utils.check_path(path) image = Image.open(path) return image + + +class TextReader: + @classmethod + def read(cls, text): + return text