Merge two or more ML models with different size of output togetger
Maybe you can use the Concatenate layer
outputs = tf.keras.Concatenate()([model1, model2])
full_model = tf.keras.Model(inputs=inputs, outputs=outputs, name='full_model')

This will simply concatenate the two softmax output into one.