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

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