I get an error when trying to use the tuned model, because apparently the last Dense layer cannot have 7 units ( I get a ValueError about how Shape (None, 1) is not compatible with (None, 7)). I changed the number of units in the Dense layer to 1 and it started working. What was happening there?