Update speech_training.py