Update README.md

README.md

@@ -1,58 +1,60 @@
----
-library_name: transformers
-tags:
-- generated_from_trainer
-model-index:
-- name: speecht5_finetuned_tr_mediaspeech_2
-  results: []
----
-
-<!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should probably proofread and complete it, then remove this comment. -->
-
-# speecht5_finetuned_tr_mediaspeech_2
-
-This model was trained from scratch on an unknown dataset.
-It achieves the following results on the evaluation set:
-- eval_loss: 0.5293
-- eval_runtime: 319.9451
-- eval_samples_per_second: 31.702
-- eval_steps_per_second: 15.853
-- epoch: 1.7388
-- step: 2000
-
-## Model description
-
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
-## Training procedure
-
-### Training hyperparameters
-
-The following hyperparameters were used during training:
-- learning_rate: 1e-05
-- train_batch_size: 4
-- eval_batch_size: 2
-- seed: 42
-- gradient_accumulation_steps: 8
-- total_train_batch_size: 32
-- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
-- lr_scheduler_type: linear
-- lr_scheduler_warmup_steps: 500
-- training_steps: 4000
-- mixed_precision_training: Native AMP
-
-### Framework versions
-
-- Transformers 4.46.3
-- Pytorch 2.5.1+cu124
-- Datasets 3.1.0
-- Tokenizers 0.20.3
+---
+library_name: transformers
+tags:
+- generated_from_trainer
+model-index:
+- name: speecht5_finetuned_tr_mediaspeech_2
+  results: []
+base_model:
+- microsoft/speecht5_tts
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# speecht5_finetuned_tr_mediaspeech_2
+
+This model was trained from scratch on an unknown dataset.
+It achieves the following results on the evaluation set:
+- eval_loss: 0.5293
+- eval_runtime: 319.9451
+- eval_samples_per_second: 31.702
+- eval_steps_per_second: 15.853
+- epoch: 1.7388
+- step: 2000
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 1e-05
+- train_batch_size: 4
+- eval_batch_size: 2
+- seed: 42
+- gradient_accumulation_steps: 8
+- total_train_batch_size: 32
+- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
+- lr_scheduler_type: linear
+- lr_scheduler_warmup_steps: 500
+- training_steps: 4000
+- mixed_precision_training: Native AMP
+
+### Framework versions
+
+- Transformers 4.46.3
+- Pytorch 2.5.1+cu124
+- Datasets 3.1.0
+- Tokenizers 0.20.3