models/__init__.py

from .generator import *
from .discriminator import *