Update train.py
Browse files
train.py
CHANGED
@@ -2,7 +2,7 @@ from datasets import load_dataset
|
|
2 |
from transformers import Trainer, TrainingArguments, Tacotron2ForConditionalGeneration
|
3 |
|
4 |
# تحميل البيانات من Hugging Face Datasets
|
5 |
-
dataset = load_dataset("
|
6 |
najdi_data = dataset.filter(lambda example: example['SpeakerDialect'] == 'Najdi')
|
7 |
|
8 |
# إعداد النموذج والمعالج
|
|
|
2 |
from transformers import Trainer, TrainingArguments, Tacotron2ForConditionalGeneration
|
3 |
|
4 |
# تحميل البيانات من Hugging Face Datasets
|
5 |
+
dataset = load_dataset("m6011/sada2022")
|
6 |
najdi_data = dataset.filter(lambda example: example['SpeakerDialect'] == 'Najdi')
|
7 |
|
8 |
# إعداد النموذج والمعالج
|