from espnet2.bin.tts_train import train
from espnet2.tasks.tts import TTSTask
from datasets import load_dataset

# تحميل البيانات للهجة النجدية
dataset = load_dataset("m6011/sada2022")
najdi_data = dataset.filter(lambda example: example['SpeakerDialect'] == 'Najdi')

# إعداد التدريب
train_config = {
    'output_dir': './results',
    'train_data_path_and_name_and_type': najdi_data['train'],
    'valid_data_path_and_name_and_type': najdi_data['test'],
    'train_batch_size': 8,
    'epochs': 10,
}

# بدء عملية التدريب باستخدام ESPnet
TTSTask.main(**train_config)