Fix imports in transformer model

1 files changed, 3 insertions, 3 deletions

diff --git a/text_recognizer/models/transformer.py b/text_recognizer/models/transformer.py
index 50bf73d..d8cb665 100644
--- a/text_recognizer/models/transformer.py
+++ b/text_recognizer/models/transformer.py
@@ -1,12 +1,12 @@
 """PyTorch Lightning model for base Transformers."""
-from typing import Tuple, Type, Set
+from typing import Set, Tuple
 
 import attr
 import torch
 from torch import Tensor
 
-from text_recognizer.models.metrics import CharacterErrorRate
 from text_recognizer.models.base import BaseLitModel
+from text_recognizer.models.metrics import CharacterErrorRate
 
 
 @attr.s(auto_attribs=True, eq=False)
@@ -76,7 +76,7 @@ class TransformerLitModel(BaseLitModel):
 
     def predict(self, x: Tensor) -> Tensor:
         """Predicts text in image.
-        
+
         Args:
             x (Tensor): Image(s) to extract text from.