ramonpeter
diff --git a/‎eventgan/__init__.py
+2-1 b/‎eventgan/__init__.py
+2-1
@@ -13,5 +13,6 @@
 
 from . import modules
 from . import utils
+from .model import *
 
-__all__ = ["modules", "utils"]
+__all__ = ["modules", "utils", "EventGAN"]