1 files changed, 6 insertions, 8 deletions
diff --git a/src/text_recognizer/models/__init__.py b/src/text_recognizer/models/__init__.py
index 28aa52e..53340f1 100644
--- a/src/text_recognizer/models/__init__.py
+++ b/src/text_recognizer/models/__init__.py
@@ -2,18 +2,16 @@
 from .base import Model
 from .character_model import CharacterModel
 from .crnn_model import CRNNModel
-from .metrics import accuracy, cer, wer
-from .transformer_encoder_model import TransformerEncoderModel
-from .vision_transformer_model import VisionTransformerModel
+from .metrics import accuracy, accuracy_ignore_pad, cer, wer
+from .transformer_model import TransformerModel
 
 __all__ = [
-    "Model",
+    "accuracy",
+    "accuracy_ignore_pad",
     "cer",
     "CharacterModel",
     "CRNNModel",
-    "CNNTransfromerModel",
-    "accuracy",
-    "TransformerEncoderModel",
-    "VisionTransformerModel",
+    "Model",
+    "TransformerModel",
     "wer",
 ]