Use keras and exported functions for this
Use keras and exported functions for this