Arbi-Houssem/Tunisian_dataset_STT-TTS1
Viewer • Updated • 831 • 32 • 1
How to use Arbi-Houssem/TunLangModel1.6 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("automatic-speech-recognition", model="Arbi-Houssem/TunLangModel1.6") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
processor = AutoProcessor.from_pretrained("Arbi-Houssem/TunLangModel1.6")
model = AutoModelForSpeechSeq2Seq.from_pretrained("Arbi-Houssem/TunLangModel1.6", device_map="auto")This model is a fine-tuned version of openai/whisper-small on the Tunisian_dataset_STT-TTS1 dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Wer |
|---|---|---|---|---|
| 0.565 | 10.3093 | 1000 | 2.7181 | 99.5460 |
| 0.2875 | 20.6186 | 2000 | 2.7486 | 106.3554 |
| 0.1701 | 30.9278 | 3000 | 2.8744 | 103.2425 |
| 0.1375 | 41.2371 | 4000 | 2.9262 | 99.4163 |
Base model
openai/whisper-small