keras-hub-nightly 0.15.0.dev20240823171555__py3-none-any.whl → 0.16.0.dev2024092017__py3-none-any.whl

keras_hub/__init__.py

@@ -11,12 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-"""DO NOT EDIT.
-
-This file was autogenerated. Do not edit it by hand,
-since your modifications would be overwritten.
-"""
-
 import os
 
 # sentencepiece segfaults on some version of tensorflow if tf is imported first.

keras_hub/api/__init__.py

@@ -17,11 +17,13 @@ This file was autogenerated. Do not edit it by hand,
 since your modifications would be overwritten.
 """
 
+from keras_hub.api import bounding_box
 from keras_hub.api import layers
 from keras_hub.api import metrics
 from keras_hub.api import models
 from keras_hub.api import samplers
 from keras_hub.api import tokenizers
+from keras_hub.api import utils
 from keras_hub.src.utils.preset_utils import upload_preset
 from keras_hub.src.version_utils import __version__
 from keras_hub.src.version_utils import version

keras_hub/api/bounding_box/__init__.py

@@ -0,0 +1,36 @@
+# Copyright 2024 The KerasHub Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""DO NOT EDIT.
+
+This file was autogenerated. Do not edit it by hand,
+since your modifications would be overwritten.
+"""
+
+from keras_hub.src.bounding_box.converters import convert_format
+from keras_hub.src.bounding_box.formats import CENTER_XYWH
+from keras_hub.src.bounding_box.formats import REL_XYWH
+from keras_hub.src.bounding_box.formats import REL_XYXY
+from keras_hub.src.bounding_box.formats import REL_YXYX
+from keras_hub.src.bounding_box.formats import XYWH
+from keras_hub.src.bounding_box.formats import XYXY
+from keras_hub.src.bounding_box.formats import YXYX
+from keras_hub.src.bounding_box.iou import compute_ciou
+from keras_hub.src.bounding_box.iou import compute_iou
+from keras_hub.src.bounding_box.to_dense import to_dense
+from keras_hub.src.bounding_box.to_ragged import to_ragged
+from keras_hub.src.bounding_box.utils import as_relative
+from keras_hub.src.bounding_box.utils import clip_boxes
+from keras_hub.src.bounding_box.utils import clip_to_image
+from keras_hub.src.bounding_box.utils import is_relative
+from keras_hub.src.bounding_box.validate_format import validate_format

keras_hub/api/layers/__init__.py

@@ -36,6 +36,8 @@ from keras_hub.src.layers.modeling.token_and_position_embedding import (
 )
 from keras_hub.src.layers.modeling.transformer_decoder import TransformerDecoder
 from keras_hub.src.layers.modeling.transformer_encoder import TransformerEncoder
+from keras_hub.src.layers.preprocessing.audio_converter import AudioConverter
+from keras_hub.src.layers.preprocessing.image_converter import ImageConverter
 from keras_hub.src.layers.preprocessing.masked_lm_mask_generator import (
     MaskedLMMaskGenerator,
 )
@@ -44,4 +46,16 @@ from keras_hub.src.layers.preprocessing.multi_segment_packer import (
 )
 from keras_hub.src.layers.preprocessing.random_deletion import RandomDeletion
 from keras_hub.src.layers.preprocessing.random_swap import RandomSwap
+from keras_hub.src.layers.preprocessing.resizing_image_converter import (
+    ResizingImageConverter,
+)
 from keras_hub.src.layers.preprocessing.start_end_packer import StartEndPacker
+from keras_hub.src.models.pali_gemma.pali_gemma_image_converter import (
+    PaliGemmaImageConverter,
+)
+from keras_hub.src.models.resnet.resnet_image_converter import (
+    ResNetImageConverter,
+)
+from keras_hub.src.models.whisper.whisper_audio_converter import (
+    WhisperAudioConverter,
+)

keras_hub/api/models/__init__.py

@@ -18,38 +18,54 @@ since your modifications would be overwritten.
 """
 
 from keras_hub.src.models.albert.albert_backbone import AlbertBackbone
-from keras_hub.src.models.albert.albert_classifier import AlbertClassifier
 from keras_hub.src.models.albert.albert_masked_lm import AlbertMaskedLM
 from keras_hub.src.models.albert.albert_masked_lm_preprocessor import (
     AlbertMaskedLMPreprocessor,
 )
-from keras_hub.src.models.albert.albert_preprocessor import AlbertPreprocessor
+from keras_hub.src.models.albert.albert_text_classifier import (
+    AlbertTextClassifier,
+)
+from keras_hub.src.models.albert.albert_text_classifier import (
+    AlbertTextClassifier as AlbertClassifier,
+)
+from keras_hub.src.models.albert.albert_text_classifier_preprocessor import (
+    AlbertTextClassifierPreprocessor,
+)
+from keras_hub.src.models.albert.albert_text_classifier_preprocessor import (
+    AlbertTextClassifierPreprocessor as AlbertPreprocessor,
+)
 from keras_hub.src.models.albert.albert_tokenizer import AlbertTokenizer
 from keras_hub.src.models.backbone import Backbone
 from keras_hub.src.models.bart.bart_backbone import BartBackbone
-from keras_hub.src.models.bart.bart_preprocessor import BartPreprocessor
 from keras_hub.src.models.bart.bart_seq_2_seq_lm import BartSeq2SeqLM
 from keras_hub.src.models.bart.bart_seq_2_seq_lm_preprocessor import (
     BartSeq2SeqLMPreprocessor,
 )
 from keras_hub.src.models.bart.bart_tokenizer import BartTokenizer
 from keras_hub.src.models.bert.bert_backbone import BertBackbone
-from keras_hub.src.models.bert.bert_classifier import BertClassifier
 from keras_hub.src.models.bert.bert_masked_lm import BertMaskedLM
 from keras_hub.src.models.bert.bert_masked_lm_preprocessor import (
     BertMaskedLMPreprocessor,
 )
-from keras_hub.src.models.bert.bert_preprocessor import BertPreprocessor
+from keras_hub.src.models.bert.bert_text_classifier import BertTextClassifier
+from keras_hub.src.models.bert.bert_text_classifier import (
+    BertTextClassifier as BertClassifier,
+)
+from keras_hub.src.models.bert.bert_text_classifier_preprocessor import (
+    BertTextClassifierPreprocessor,
+)
+from keras_hub.src.models.bert.bert_text_classifier_preprocessor import (
+    BertTextClassifierPreprocessor as BertPreprocessor,
+)
 from keras_hub.src.models.bert.bert_tokenizer import BertTokenizer
 from keras_hub.src.models.bloom.bloom_backbone import BloomBackbone
 from keras_hub.src.models.bloom.bloom_causal_lm import BloomCausalLM
 from keras_hub.src.models.bloom.bloom_causal_lm_preprocessor import (
     BloomCausalLMPreprocessor,
 )
-from keras_hub.src.models.bloom.bloom_preprocessor import BloomPreprocessor
 from keras_hub.src.models.bloom.bloom_tokenizer import BloomTokenizer
 from keras_hub.src.models.causal_lm import CausalLM
-from keras_hub.src.models.classifier import Classifier
+from keras_hub.src.models.causal_lm_preprocessor import CausalLMPreprocessor
 from keras_hub.src.models.csp_darknet.csp_darknet_backbone import (
     CSPDarkNetBackbone,
 )
@@ -59,17 +75,23 @@ from keras_hub.src.models.csp_darknet.csp_darknet_image_classifier import (
 from keras_hub.src.models.deberta_v3.deberta_v3_backbone import (
     DebertaV3Backbone,
 )
-from keras_hub.src.models.deberta_v3.deberta_v3_classifier import (
-    DebertaV3Classifier,
-)
 from keras_hub.src.models.deberta_v3.deberta_v3_masked_lm import (
     DebertaV3MaskedLM,
 )
 from keras_hub.src.models.deberta_v3.deberta_v3_masked_lm_preprocessor import (
     DebertaV3MaskedLMPreprocessor,
 )
-from keras_hub.src.models.deberta_v3.deberta_v3_preprocessor import (
-    DebertaV3Preprocessor,
+from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier import (
+    DebertaV3TextClassifier,
+)
+from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier import (
+    DebertaV3TextClassifier as DebertaV3Classifier,
+)
+from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier_preprocessor import (
+    DebertaV3TextClassifierPreprocessor,
+)
+from keras_hub.src.models.deberta_v3.deberta_v3_text_classifier_preprocessor import (
+    DebertaV3TextClassifierPreprocessor as DebertaV3Preprocessor,
 )
 from keras_hub.src.models.deberta_v3.deberta_v3_tokenizer import (
     DebertaV3Tokenizer,
@@ -81,40 +103,53 @@ from keras_hub.src.models.densenet.densenet_image_classifier import (
 from keras_hub.src.models.distil_bert.distil_bert_backbone import (
     DistilBertBackbone,
 )
-from keras_hub.src.models.distil_bert.distil_bert_classifier import (
-    DistilBertClassifier,
-)
 from keras_hub.src.models.distil_bert.distil_bert_masked_lm import (
     DistilBertMaskedLM,
 )
 from keras_hub.src.models.distil_bert.distil_bert_masked_lm_preprocessor import (
     DistilBertMaskedLMPreprocessor,
 )
-from keras_hub.src.models.distil_bert.distil_bert_preprocessor import (
-    DistilBertPreprocessor,
+from keras_hub.src.models.distil_bert.distil_bert_text_classifier import (
+    DistilBertTextClassifier,
+)
+from keras_hub.src.models.distil_bert.distil_bert_text_classifier import (
+    DistilBertTextClassifier as DistilBertClassifier,
+)
+from keras_hub.src.models.distil_bert.distil_bert_text_classifier_preprocessor import (
+    DistilBertTextClassifierPreprocessor,
+)
+from keras_hub.src.models.distil_bert.distil_bert_text_classifier_preprocessor import (
+    DistilBertTextClassifierPreprocessor as DistilBertPreprocessor,
 )
 from keras_hub.src.models.distil_bert.distil_bert_tokenizer import (
     DistilBertTokenizer,
 )
-from keras_hub.src.models.electra.electra_backbone import ElectraBackbone
-from keras_hub.src.models.electra.electra_preprocessor import (
-    ElectraPreprocessor,
+from keras_hub.src.models.efficientnet.efficientnet_backbone import (
+    EfficientNetBackbone,
 )
+from keras_hub.src.models.electra.electra_backbone import ElectraBackbone
 from keras_hub.src.models.electra.electra_tokenizer import ElectraTokenizer
 from keras_hub.src.models.f_net.f_net_backbone import FNetBackbone
-from keras_hub.src.models.f_net.f_net_classifier import FNetClassifier
 from keras_hub.src.models.f_net.f_net_masked_lm import FNetMaskedLM
 from keras_hub.src.models.f_net.f_net_masked_lm_preprocessor import (
     FNetMaskedLMPreprocessor,
 )
-from keras_hub.src.models.f_net.f_net_preprocessor import FNetPreprocessor
+from keras_hub.src.models.f_net.f_net_text_classifier import FNetTextClassifier
+from keras_hub.src.models.f_net.f_net_text_classifier import (
+    FNetTextClassifier as FNetClassifier,
+)
+from keras_hub.src.models.f_net.f_net_text_classifier_preprocessor import (
+    FNetTextClassifierPreprocessor,
+)
+from keras_hub.src.models.f_net.f_net_text_classifier_preprocessor import (
+    FNetTextClassifierPreprocessor as FNetPreprocessor,
+)
 from keras_hub.src.models.f_net.f_net_tokenizer import FNetTokenizer
 from keras_hub.src.models.falcon.falcon_backbone import FalconBackbone
 from keras_hub.src.models.falcon.falcon_causal_lm import FalconCausalLM
 from keras_hub.src.models.falcon.falcon_causal_lm_preprocessor import (
     FalconCausalLMPreprocessor,
 )
-from keras_hub.src.models.falcon.falcon_preprocessor import FalconPreprocessor
 from keras_hub.src.models.falcon.falcon_tokenizer import FalconTokenizer
 from keras_hub.src.models.feature_pyramid_backbone import FeaturePyramidBackbone
 from keras_hub.src.models.gemma.gemma_backbone import GemmaBackbone
@@ -122,7 +157,6 @@ from keras_hub.src.models.gemma.gemma_causal_lm import GemmaCausalLM
 from keras_hub.src.models.gemma.gemma_causal_lm_preprocessor import (
     GemmaCausalLMPreprocessor,
 )
-from keras_hub.src.models.gemma.gemma_preprocessor import GemmaPreprocessor
 from keras_hub.src.models.gemma.gemma_tokenizer import GemmaTokenizer
 from keras_hub.src.models.gpt2.gpt2_backbone import GPT2Backbone
 from keras_hub.src.models.gpt2.gpt2_causal_lm import GPT2CausalLM
@@ -136,34 +170,30 @@ from keras_hub.src.models.gpt_neo_x.gpt_neo_x_causal_lm import GPTNeoXCausalLM
 from keras_hub.src.models.gpt_neo_x.gpt_neo_x_causal_lm_preprocessor import (
     GPTNeoXCausalLMPreprocessor,
 )
-from keras_hub.src.models.gpt_neo_x.gpt_neo_x_preprocessor import (
-    GPTNeoXPreprocessor,
-)
 from keras_hub.src.models.gpt_neo_x.gpt_neo_x_tokenizer import GPTNeoXTokenizer
 from keras_hub.src.models.image_classifier import ImageClassifier
+from keras_hub.src.models.image_classifier_preprocessor import (
+    ImageClassifierPreprocessor,
+)
 from keras_hub.src.models.llama3.llama3_backbone import Llama3Backbone
 from keras_hub.src.models.llama3.llama3_causal_lm import Llama3CausalLM
 from keras_hub.src.models.llama3.llama3_causal_lm_preprocessor import (
     Llama3CausalLMPreprocessor,
 )
-from keras_hub.src.models.llama3.llama3_preprocessor import Llama3Preprocessor
 from keras_hub.src.models.llama3.llama3_tokenizer import Llama3Tokenizer
 from keras_hub.src.models.llama.llama_backbone import LlamaBackbone
 from keras_hub.src.models.llama.llama_causal_lm import LlamaCausalLM
 from keras_hub.src.models.llama.llama_causal_lm_preprocessor import (
     LlamaCausalLMPreprocessor,
 )
-from keras_hub.src.models.llama.llama_preprocessor import LlamaPreprocessor
 from keras_hub.src.models.llama.llama_tokenizer import LlamaTokenizer
 from keras_hub.src.models.masked_lm import MaskedLM
+from keras_hub.src.models.masked_lm_preprocessor import MaskedLMPreprocessor
 from keras_hub.src.models.mistral.mistral_backbone import MistralBackbone
 from keras_hub.src.models.mistral.mistral_causal_lm import MistralCausalLM
 from keras_hub.src.models.mistral.mistral_causal_lm_preprocessor import (
     MistralCausalLMPreprocessor,
 )
-from keras_hub.src.models.mistral.mistral_preprocessor import (
-    MistralPreprocessor,
-)
 from keras_hub.src.models.mistral.mistral_tokenizer import MistralTokenizer
 from keras_hub.src.models.mix_transformer.mix_transformer_backbone import (
     MiTBackbone,
@@ -171,12 +201,15 @@ from keras_hub.src.models.mix_transformer.mix_transformer_backbone import (
 from keras_hub.src.models.mix_transformer.mix_transformer_classifier import (
     MiTImageClassifier,
 )
+from keras_hub.src.models.mobilenet.mobilenet_backbone import MobileNetBackbone
+from keras_hub.src.models.mobilenet.mobilenet_image_classifier import (
+    MobileNetImageClassifier,
+)
 from keras_hub.src.models.opt.opt_backbone import OPTBackbone
 from keras_hub.src.models.opt.opt_causal_lm import OPTCausalLM
 from keras_hub.src.models.opt.opt_causal_lm_preprocessor import (
     OPTCausalLMPreprocessor,
 )
-from keras_hub.src.models.opt.opt_preprocessor import OPTPreprocessor
 from keras_hub.src.models.opt.opt_tokenizer import OPTTokenizer
 from keras_hub.src.models.pali_gemma.pali_gemma_backbone import (
     PaliGemmaBackbone,
@@ -195,52 +228,68 @@ from keras_hub.src.models.phi3.phi3_causal_lm import Phi3CausalLM
 from keras_hub.src.models.phi3.phi3_causal_lm_preprocessor import (
     Phi3CausalLMPreprocessor,
 )
-from keras_hub.src.models.phi3.phi3_preprocessor import Phi3Preprocessor
 from keras_hub.src.models.phi3.phi3_tokenizer import Phi3Tokenizer
 from keras_hub.src.models.preprocessor import Preprocessor
 from keras_hub.src.models.resnet.resnet_backbone import ResNetBackbone
 from keras_hub.src.models.resnet.resnet_image_classifier import (
     ResNetImageClassifier,
 )
+from keras_hub.src.models.resnet.resnet_image_classifier_preprocessor import (
+    ResNetImageClassifierPreprocessor,
+)
 from keras_hub.src.models.roberta.roberta_backbone import RobertaBackbone
-from keras_hub.src.models.roberta.roberta_classifier import RobertaClassifier
 from keras_hub.src.models.roberta.roberta_masked_lm import RobertaMaskedLM
 from keras_hub.src.models.roberta.roberta_masked_lm_preprocessor import (
     RobertaMaskedLMPreprocessor,
 )
-from keras_hub.src.models.roberta.roberta_preprocessor import (
-    RobertaPreprocessor,
+from keras_hub.src.models.roberta.roberta_text_classifier import (
+    RobertaTextClassifier,
+)
+from keras_hub.src.models.roberta.roberta_text_classifier import (
+    RobertaTextClassifier as RobertaClassifier,
+)
+from keras_hub.src.models.roberta.roberta_text_classifier_preprocessor import (
+    RobertaTextClassifierPreprocessor,
+)
+from keras_hub.src.models.roberta.roberta_text_classifier_preprocessor import (
+    RobertaTextClassifierPreprocessor as RobertaPreprocessor,
 )
 from keras_hub.src.models.roberta.roberta_tokenizer import RobertaTokenizer
 from keras_hub.src.models.seq_2_seq_lm import Seq2SeqLM
+from keras_hub.src.models.seq_2_seq_lm_preprocessor import Seq2SeqLMPreprocessor
 from keras_hub.src.models.t5.t5_backbone import T5Backbone
 from keras_hub.src.models.t5.t5_tokenizer import T5Tokenizer
 from keras_hub.src.models.task import Task
+from keras_hub.src.models.text_classifier import TextClassifier
+from keras_hub.src.models.text_classifier import TextClassifier as Classifier
+from keras_hub.src.models.text_classifier_preprocessor import (
+    TextClassifierPreprocessor,
+)
 from keras_hub.src.models.vgg.vgg_backbone import VGGBackbone
 from keras_hub.src.models.vgg.vgg_image_classifier import VGGImageClassifier
 from keras_hub.src.models.vit_det.vit_det_backbone import ViTDetBackbone
-from keras_hub.src.models.whisper.whisper_audio_feature_extractor import (
-    WhisperAudioFeatureExtractor,
-)
 from keras_hub.src.models.whisper.whisper_backbone import WhisperBackbone
-from keras_hub.src.models.whisper.whisper_preprocessor import (
-    WhisperPreprocessor,
-)
 from keras_hub.src.models.whisper.whisper_tokenizer import WhisperTokenizer
 from keras_hub.src.models.xlm_roberta.xlm_roberta_backbone import (
     XLMRobertaBackbone,
 )
-from keras_hub.src.models.xlm_roberta.xlm_roberta_classifier import (
-    XLMRobertaClassifier,
-)
 from keras_hub.src.models.xlm_roberta.xlm_roberta_masked_lm import (
     XLMRobertaMaskedLM,
 )
 from keras_hub.src.models.xlm_roberta.xlm_roberta_masked_lm_preprocessor import (
     XLMRobertaMaskedLMPreprocessor,
 )
-from keras_hub.src.models.xlm_roberta.xlm_roberta_preprocessor import (
-    XLMRobertaPreprocessor,
+from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier import (
+    XLMRobertaTextClassifier,
+)
+from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier import (
+    XLMRobertaTextClassifier as XLMRobertaClassifier,
+)
+from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier_preprocessor import (
+    XLMRobertaTextClassifierPreprocessor,
+)
+from keras_hub.src.models.xlm_roberta.xlm_roberta_text_classifier_preprocessor import (
+    XLMRobertaTextClassifierPreprocessor as XLMRobertaPreprocessor,
 )
 from keras_hub.src.models.xlm_roberta.xlm_roberta_tokenizer import (
     XLMRobertaTokenizer,

keras_hub/api/tokenizers/__init__.py

@@ -17,6 +17,36 @@ This file was autogenerated. Do not edit it by hand,
 since your modifications would be overwritten.
 """
 
+from keras_hub.src.models.albert.albert_tokenizer import AlbertTokenizer
+from keras_hub.src.models.bart.bart_tokenizer import BartTokenizer
+from keras_hub.src.models.bert.bert_tokenizer import BertTokenizer
+from keras_hub.src.models.bloom.bloom_tokenizer import BloomTokenizer
+from keras_hub.src.models.deberta_v3.deberta_v3_tokenizer import (
+    DebertaV3Tokenizer,
+)
+from keras_hub.src.models.distil_bert.distil_bert_tokenizer import (
+    DistilBertTokenizer,
+)
+from keras_hub.src.models.electra.electra_tokenizer import ElectraTokenizer
+from keras_hub.src.models.f_net.f_net_tokenizer import FNetTokenizer
+from keras_hub.src.models.falcon.falcon_tokenizer import FalconTokenizer
+from keras_hub.src.models.gemma.gemma_tokenizer import GemmaTokenizer
+from keras_hub.src.models.gpt2.gpt2_tokenizer import GPT2Tokenizer
+from keras_hub.src.models.gpt_neo_x.gpt_neo_x_tokenizer import GPTNeoXTokenizer
+from keras_hub.src.models.llama3.llama3_tokenizer import Llama3Tokenizer
+from keras_hub.src.models.llama.llama_tokenizer import LlamaTokenizer
+from keras_hub.src.models.mistral.mistral_tokenizer import MistralTokenizer
+from keras_hub.src.models.opt.opt_tokenizer import OPTTokenizer
+from keras_hub.src.models.pali_gemma.pali_gemma_tokenizer import (
+    PaliGemmaTokenizer,
+)
+from keras_hub.src.models.phi3.phi3_tokenizer import Phi3Tokenizer
+from keras_hub.src.models.roberta.roberta_tokenizer import RobertaTokenizer
+from keras_hub.src.models.t5.t5_tokenizer import T5Tokenizer
+from keras_hub.src.models.whisper.whisper_tokenizer import WhisperTokenizer
+from keras_hub.src.models.xlm_roberta.xlm_roberta_tokenizer import (
+    XLMRobertaTokenizer,
+)
 from keras_hub.src.tokenizers.byte_pair_tokenizer import BytePairTokenizer
 from keras_hub.src.tokenizers.byte_tokenizer import ByteTokenizer
 from keras_hub.src.tokenizers.sentence_piece_tokenizer import (

keras_hub/api/utils/__init__.py

@@ -0,0 +1,22 @@
+# Copyright 2024 The KerasHub Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""DO NOT EDIT.
+
+This file was autogenerated. Do not edit it by hand,
+since your modifications would be overwritten.
+"""
+
+from keras_hub.src.utils.imagenet.imagenet_utils import (
+    decode_imagenet_predictions,
+)

keras_hub/src/api_export.py

@@ -14,7 +14,7 @@
 
 import types
 
-import keras
+from keras.saving import register_keras_serializable
 
 try:
     import namex
@@ -22,14 +22,20 @@ except ImportError:
     namex = None
 
 
-def maybe_register_serializable(symbol):
+def maybe_register_serializable(path, symbol):
+    if isinstance(path, (list, tuple)):
+        # If we have multiple export names, actually make sure to register these
+        # first. This makes sure we have a backward compat mapping of old
+        # serialized names to new class.
+        for name in path:
+            name = name.split(".")[-1]
+            register_keras_serializable(package="keras_nlp", name=name)(symbol)
+            register_keras_serializable(package="keras_hub", name=name)(symbol)
     if isinstance(symbol, types.FunctionType) or hasattr(symbol, "get_config"):
-        # We register twice, first with the old name, second with the new name,
-        # so loading still works under the old name.
-        # TODO replace keras_nlp with keras-hub after rename.
-        compat_name = "keras_nlp"
-        keras.saving.register_keras_serializable(package=compat_name)(symbol)
-        keras.saving.register_keras_serializable(package="keras_hub")(symbol)
+        # We register twice, first with keras_nlp, second with keras_hub,
+        # so loading still works for classes saved as "keras_nlp".
+        register_keras_serializable(package="keras_nlp")(symbol)
+        register_keras_serializable(package="keras_hub")(symbol)
 
 
 if namex:
@@ -39,7 +45,7 @@ if namex:
             super().__init__(package="keras_hub", path=path)
 
         def __call__(self, symbol):
-            maybe_register_serializable(symbol)
+            maybe_register_serializable(self.path, symbol)
             return super().__call__(symbol)
 
 else:

keras_hub/src/bounding_box/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 2024 The KerasHub Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.