# import the necessary packages from tensorflow import keras def loader(stem, trunk, attn): # load the models from disk conv_stem = keras.models.load_model( stem, compile=False ) conv_trunk = keras.models.load_model( trunk, compile=False ) conv_attn = keras.models.load_model( attn, compile=False ) return (conv_stem, conv_trunk, conv_attn)