Import register_model and get_model_function
@@ -2,6 +2,7 @@ from .language import PyTT_Language
from .tok2vec import PyTT_TokenVectorEncoder # noqa
from .pipeline import PyTT_TextCategorizer # noqa
from .wordpiecer import PyTT_WordPiecer # noqa
+from .model_registry import register_model, get_model_function # noqa
from .about import __version__ # noqa
PyTT_Language.install_extensions()
