def make_model():
vae = VAE()
vae.set_weights('./vae/weights.h5')
rnn = RNN()
rnn.set_weights('./rnn/weights.h5')
controller = Controller()
model = Model(controller, vae, rnn)
return model
def make_model():
acai = ACAI()
rnn = RNN()
rnn.set_weights('./rnn/weights.h5')
controller = Controller()
model = Model(controller, acai, rnn)
return model