Add LM-boosted decoding

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, Wav2Vec2ProcessorWithLM
 from librosa import load, resample
 
 
 asr_model = 'facebook/wav2vec2-base-960h'
-
-asr = pipeline('automatic-speech-recognition', model=asr_model, feature_extractor=asr_model)
+processor = Wav2Vec2ProcessorWithLM.from_pretrained(asr_model)
+asr = pipeline('automatic-speech-recognition', model=asr_model, tokenizer=processor.tokenizer, feature_extractor=processor.feature_extractor, decoder=processor.decoder)
 
 def transcribe(filepath):
 	speech, sampling_rate = load(filepath)