# Just plug in your Keras model
train = tff.learning.build_federated_averaging_process(...)