my_model = models.Sequential()
pre_model = pretrain_model(params)
my_model.add(pre_model)
my_model.add(layers.Flatten())
my_model.add(layers.Dense(1024, activation="relu"))
my_model.add(layers.Dense(512, activation="relu"))
my_model.add(layers.Dropout(params["Dropout"]))
my_model.add(layers.Dense(
len(params["Classes"]), activation="softmax"))
return my_model