register parseq preset (#2438)

* register parseq preset

* rename preset

Author: sachinprasadhs

keras_hub/src/models/parseq/__init__.py

@@ -0,0 +1,5 @@
+from keras_hub.src.models.parseq.parseq_backbone import PARSeqBackbone
+from keras_hub.src.models.parseq.parseq_presets import backbone_presets
+from keras_hub.src.utils.preset_utils import register_presets
+
+register_presets(backbone_presets, PARSeqBackbone)

keras_hub/src/models/parseq/parseq_presets.py

@@ -0,0 +1,15 @@
+"""PARSeq preset configurations."""
+
+backbone_presets = {
+    "parseq": {
+        "metadata": {
+            "description": (
+                "Permuted autoregressive sequence (PARSeq) base "
+                "model for scene text recognition"
+            ),
+            "params": 23_832_671,
+            "path": "parseq",
+        },
+        "kaggle_handle": "kaggle://keras/parseq/keras/parseq/1",
+    }
+}

tools/checkpoint_conversion/convert_parseq_checkpoints.py

@@ -33,7 +33,7 @@
 
 FLAGS = flags.FLAGS
 
-PRESET_MAP = {"parseq_vit": "baudm/parseq"}
+PRESET_MAP = {"parseq": "baudm/parseq"}
 
 flags.DEFINE_string(
     "preset",
@@ -336,7 +336,7 @@ def main(_):
 
     validate_output(parseq_preprocessor, keras_model, torch_model)
     print("✅ Outputs Validated.")
-
+    keras_model.save_to_preset(f"./{preset}")
     print(f"🏁 Preset saved to ./{preset}.")
 
     upload_uri = FLAGS.upload_uri