Okay, it seems that you have a phantom dimension that you need to get rid of.
The tensor inputs has shape [batch, 1, 224], you need to squeeze it to have [batch, 224] instead.
Instead of calling outputs = model(inputs), try :
outputs = model(inputs.squeeze(1))