utilities/load_model.py

# 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)