flozi00 commited on
Commit
6e3264b
·
1 Parent(s): ee00afc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,9 +4,9 @@ import re
4
  import torch
5
  from pyctcdecode import BeamSearchDecoderCTC
6
 
7
- lmID = "aware-ai/german-lowercase-4gram-kenlm"
8
  decoder = BeamSearchDecoderCTC.load_from_hf_hub(lmID)
9
- p = pipeline("automatic-speech-recognition", model="aware-ai/robust-wav2vec2-base-german-lowercase", decoder=decoder)
10
 
11
  model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_1.2B")
12
  tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_1.2B")
 
4
  import torch
5
  from pyctcdecode import BeamSearchDecoderCTC
6
 
7
+ lmID = "aware-ai/german-lowercase-5gram-kenlm"
8
  decoder = BeamSearchDecoderCTC.load_from_hf_hub(lmID)
9
+ p = pipeline("automatic-speech-recognition", model="aware-ai/robust-wav2vec2-xls-r-300m-german-lowercase", decoder=decoder)
10
 
11
  model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_1.2B")
12
  tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_1.2B")