From 4d7713746eb936832e84852e90292936b933e87d Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Thu, 22 Oct 2020 22:45:58 +0200
Subject: Transfomer added, many other changes.

---
 src/text_recognizer/models/__init__.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(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 a3cfc15..0855079 100644
--- a/src/text_recognizer/models/__init__.py
+++ b/src/text_recognizer/models/__init__.py
@@ -3,5 +3,14 @@ from .base import Model
 from .character_model import CharacterModel
 from .line_ctc_model import LineCTCModel
 from .metrics import accuracy, cer, wer
+from .vision_transformer_model import VisionTransformerModel
 
-__all__ = ["Model", "cer", "CharacterModel", "LineCTCModel", "accuracy", "wer"]
+__all__ = [
+    "Model",
+    "cer",
+    "CharacterModel",
+    "CNNTransfromerModel",
+    "LineCTCModel",
+    "accuracy",
+    "wer",
+]
-- 
cgit v1.2.3-70-g09d2