Fix keras

With upgrading the TF version to 2.5.0, unfortunately the current way Keras is used is not working anymore. So this MR imports Keras as a backend of TF. This seems to fix the issue. But maybe all models that are already trained needs to be retrained due to compatibility issues (not sure about that. Need to test it).