diff --git a/scripts/hub_master.py b/scripts/hub_master.py index a00fde0dd0..7e5f576b4f 100644 --- a/scripts/hub_master.py +++ b/scripts/hub_master.py @@ -655,6 +655,48 @@ }, ], }, + { + "path": "deit/", + "title": "DeiT", + "toc": True, + "children": [ + { + "path": "deit_image_converter", + "title": "DeiTImageConverter", + "generate": [ + "keras_hub.layers.DeiTImageConverter", + "keras_hub.layers.DeiTImageConverter.from_preset", + ], + }, + { + "path": "deit_backbone", + "title": "DeiTBackbone model", + "generate": [ + "keras_hub.models.DeiTBackbone", + "keras_hub.models.DeiTBackbone.from_preset", + ], + }, + { + "path": "deit_image_classifier", + "title": "DeiTImageClassifier model", + "generate": [ + "keras_hub.models.DeiTImageClassifier", + "keras_hub.models.DeiTImageClassifier.from_preset", + "keras_hub.models.DeiTImageClassifier.backbone", + "keras_hub.models.DeiTImageClassifier.preprocessor", + ], + }, + { + "path": "deit_image_classifier_preprocessor", + "title": "DeiTImageClassifierPreprocessor layer", + "generate": [ + "keras_hub.models.DeiTImageClassifierPreprocessor", + "keras_hub.models.DeiTImageClassifierPreprocessor.from_preset", + "keras_hub.models.DeiTImageClassifierPreprocessor.image_converter", + ], + }, + ], + }, { "path": "densenet/", "title": "DenseNet",