src/models/__init__.py

from .generator import Generator
from .discriminator import Discriminator

__all__ = ["Generator", "Discriminator"]