Weights_Loading.py

from Imports import *

model_state = joblib.load('tacotron2_inference_ema.pkl')
vocoder_state = joblib.load('vocoder_inference.pkl')

if model_state is not None:
    print("Model state loaded successfully")
else:
    print("Model state not loaded")