from .Model import Model
from .autoencoder import AutoEncoder
from .cnn import CNNPredictor
from .transformer import ByteTransformer

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