Update app.py

app.py

@@ -2,8 +2,11 @@ from transformers import pipeline, M2M100ForConditionalGeneration, M2M100Tokeniz
 import gradio as gr
 import re
 import torch
+from pyctcdecode import BeamSearchDecoderCTC
 
-p = pipeline("automatic-speech-recognition", model="aware-ai/robust-wav2vec2-base-german-lowercase")
+lmID = "aware-ai/german-lowercase-5gram-kenlm"
+decoder = BeamSearchDecoderCTC.load_from_hf_hub(lmID)
+p = pipeline("automatic-speech-recognition", model="aware-ai/robust-wav2vec2-base-german-lowercase", tokenizer=lmID, feature_extractor=lmID, decoder=decoder)
 ttp = pipeline("text2text-generation", model="aware-ai/marian-german-grammar")
 
 model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_1.2B")