TensorFlow Simple audio recognition: Can not squeeze dim[1], expected a dimension of 1, got 2

Question

TensorFlow: 2.1.0

I am trying to make my own audio classifier using TensorFlow\'s examlpe, found here. I\'m slowly making my way through the example

I aim for my c