From 7e8e54e84c63171e748bbf09516fd517e6821ace Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sat, 20 Mar 2021 18:09:06 +0100
Subject: Inital commit for refactoring to lightning

---
 text_recognizer/models/__init__.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 text_recognizer/models/__init__.py

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

diff --git a/text_recognizer/models/__init__.py b/text_recognizer/models/__init__.py
new file mode 100644
index 0000000..7647d7e
--- /dev/null
+++ b/text_recognizer/models/__init__.py
@@ -0,0 +1,18 @@
+"""Model modules."""
+from .base import Model
+from .character_model import CharacterModel
+from .crnn_model import CRNNModel
+from .ctc_transformer_model import CTCTransformerModel
+from .segmentation_model import SegmentationModel
+from .transformer_model import TransformerModel
+from .vqvae_model import VQVAEModel
+
+__all__ = [
+    "CharacterModel",
+    "CRNNModel",
+    "CTCTransformerModel",
+    "Model",
+    "SegmentationModel",
+    "TransformerModel",
+    "VQVAEModel",
+]
-- 
cgit v1.2.3-70-g09d2