federated learning on a baby cry dataset, along with a gradio frontend for accessibility to test the model