from Tacotron_V2_Layers import *
from Vocoder_Layers import *

encoder_model = build_encoder()
decoder_model = build_tacotron2()
postnet_model = build_postnet()       
vocoder       = build_generator((None, 80))

if encoder_model is not None:
    print("Encoder model loaded successfully")
else:
    print("Encoder model not loaded")
if decoder_model is not None:
    print("Decoder model loaded successfully")
else:
    print("Decoder model not loaded")
if postnet_model is not None:
    print("Postnet model loaded successfully")
else:
    print("Postnet model not loaded")
if vocoder is not None:
    print("Vocoder model loaded successfully")
else:
    print("Vocoder model not loaded")