medfadiabaidi commited on
Commit
a0df215
·
verified ·
1 Parent(s): 4298d8f

Add model card with evaluation metrics

Browse files
Files changed (1) hide show
  1. README.md +64 -0
README.md ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - ar
4
+ - tn
5
+ tags:
6
+ - automatic-speech-recognition
7
+ - whisper
8
+ - tunisian
9
+ - generated_from_trainer
10
+ dataset:
11
+ - fbougares/TEDxTN
12
+ model-index:
13
+ - name: Whisper Small Tunisian
14
+ results:
15
+ - task:
16
+ name: Automatic Speech Recognition
17
+ type: automatic-speech-recognition
18
+ dataset:
19
+ name: TEDxTN
20
+ type: fbougares/TEDxTN
21
+ config: default
22
+ split: test
23
+ metrics:
24
+ - name: WER
25
+ type: wer
26
+ value: 37.99
27
+ - name: CER
28
+ type: cer
29
+ value: 18.77
30
+ ---
31
+
32
+ # Whisper Small Fine-tuned on Tunisian Dialect (TEDxTN)
33
+
34
+ This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the [TEDxTN dataset](https://huggingface.co/datasets/fbougares/TEDxTN). It was trained to transcribe Tunisian dialect (Derja/Arabizi).
35
+
36
+ ## Model Description
37
+ - **Model:** openai/whisper-small
38
+ - **Language:** Tunisian Arabic (Derja)
39
+ - **Dataset:** TEDxTN (~22 hours)
40
+
41
+ ## Evaluation Results
42
+ | Metric | Score |
43
+ | :--- | :--- |
44
+ | **WER** | 37.99% |
45
+ | **CER** | 18.77% |
46
+
47
+ ## Usage
48
+ ```python
49
+ from transformers import WhisperForConditionalGeneration, WhisperProcessor
50
+
51
+ model_id = "medfadiabaidi/whisper-small-tunisian-asr"
52
+ processor = WhisperProcessor.from_pretrained(model_id)
53
+ model = WhisperForConditionalGeneration.from_pretrained(model_id)
54
+
55
+ # audio_input = ... # Load your audio here
56
+ # inputs = processor(audio_input, return_tensors="pt")
57
+ # generated_ids = model.generate(inputs.input_features)
58
+ # transcription = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
59
+ ```
60
+
61
+ ## Training Details
62
+ - **Epochs:** 15
63
+ - **Batch Size:** 64
64
+ - **Learning Rate:** 1e-05