diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..6ccc6b8
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,3 @@
+from .model_info_2d import model_info_2d
+
+__all__ = ["model_info_2d"]
\ No newline at end of file