from tensorflow.keras import Input, Model, Sequential
from tensorflow.keras.layers import LeakyReLU, BatchNormalization, Dense


def build_generator(latent_dim):
    model = Sequential()
    model.add(Dense(64, input_dim=latent_dim, kernel_initializer='glorot_uniform'))
    model.add(LeakyReLU(alpha=0.1))
    model.add(BatchNormalization(momentum=0.8))
    model.add(Dense(128, kernel_initializer='glorot_uniform'))
    model.add(LeakyReLU(alpha=0.1))
    model.add(BatchNormalization(momentum=0.8))
    model.add(Dense(121, kernel_initializer='glorot_uniform'))
    model.summary()
    noise = Input(shape=(latent_dim,))
    fake = model(noise)
    return Model(noise, fake)