I\'m trying to do use a part of a pretrained model to make a new model, but have some difficulties in adapting the inputs of the models. Here is the inputs and main structure of
