Trying to get the CNNTransformer to work, but it is hard.

1 files changed, 5 insertions, 2 deletions

diff --git a/src/text_recognizer/models/__init__.py b/src/text_recognizer/models/__init__.py
index 0855079..28aa52e 100644
--- a/src/text_recognizer/models/__init__.py
+++ b/src/text_recognizer/models/__init__.py
@@ -1,16 +1,19 @@
 """Model modules."""
 from .base import Model
 from .character_model import CharacterModel
-from .line_ctc_model import LineCTCModel
+from .crnn_model import CRNNModel
 from .metrics import accuracy, cer, wer
+from .transformer_encoder_model import TransformerEncoderModel
 from .vision_transformer_model import VisionTransformerModel
 
 __all__ = [
     "Model",
     "cer",
     "CharacterModel",
+    "CRNNModel",
     "CNNTransfromerModel",
-    "LineCTCModel",
     "accuracy",
+    "TransformerEncoderModel",
+    "VisionTransformerModel",
     "wer",
 ]