The model architecture
This is the task that I am performing using this model
Here is my model
def create_convolution_layers(input_img): #