from .Model import Model
from .cnn import CNNPredictor
from .transformer import Transformer


model_called: dict[str, type[Model]] = {
    'cnn': CNNPredictor,
    'transformer': Transformer
}