Compiling the Keras model