Commit
·
1d6f694
1
Parent(s):
f185505
Updated alphabet
Browse files- alphabet.json +1 -3
- lm.py +0 -3
- preprocessor_config.json +4 -1
alphabet.json
CHANGED
@@ -1,3 +1 @@
|
|
1 |
-
{"labels": ["
|
2 |
-
|
3 |
-
|
|
|
1 |
+
{"labels": ["", "R", "Z", "<s>", " ", "0", "9", "G", "F", "3", "L", "E", "Y", "Å", "A", "U", "J", "X", "T", "I", "M", "P", "N", "B", "É", "7", "</s>", "Ä", "1", "4", "C", "Ö", "⁇", "H", "V", "'", "5", "K", "W", "Q", "O", "6", "2", "8", "D", "S"], "is_bpe": false}
|
|
|
|
lm.py
CHANGED
@@ -22,8 +22,5 @@ input_values = processor(resampled_audio, return_tensors="pt").input_values
|
|
22 |
|
23 |
with torch.no_grad():
|
24 |
logits = model(input_values).logits
|
25 |
-
import pdb
|
26 |
-
pdb.set_trace()
|
27 |
-
|
28 |
transcription = processor.batch_decode(logits.numpy()).text
|
29 |
print(transcription)
|
|
|
22 |
|
23 |
with torch.no_grad():
|
24 |
logits = model(input_values).logits
|
|
|
|
|
|
|
25 |
transcription = processor.batch_decode(logits.numpy()).text
|
26 |
print(transcription)
|
preprocessor_config.json
CHANGED
@@ -5,5 +5,8 @@
|
|
5 |
"padding_side": "right",
|
6 |
"padding_value": 0,
|
7 |
"return_attention_mask": true,
|
8 |
-
"sampling_rate": 16000
|
|
|
9 |
}
|
|
|
|
|
|
5 |
"padding_side": "right",
|
6 |
"padding_value": 0,
|
7 |
"return_attention_mask": true,
|
8 |
+
"sampling_rate": 16000,
|
9 |
+
"processor_class": "Wav2Vec2ProcessorWithLM"
|
10 |
}
|
11 |
+
|
12 |
+
|