From beeaef529e7c893a3475fe27edc880e283373725 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Sun, 8 Nov 2020 12:41:04 +0100
Subject: Trying to get the CNNTransformer to work, but it is hard.

---
 src/text_recognizer/models/__init__.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'src/text_recognizer/models/__init__.py')

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",
 ]
-- 
cgit v1.2.3-70-g09d2