Alvin-Nahabwe commited on
Commit
d045f90
·
verified ·
1 Parent(s): e6ca711

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def transcribe(audio, language):
32
  # load processor
33
  p = Wav2Vec2ProcessorWithLM.from_pretrained(model_map[language], revision=revision)
34
  # load eval pipeline
35
- asr = pipeline("automatic-speech-recognition", model=model_map[language], tokenizer=p.tokenizer, feature_extractor=p.feature_extractor, decoder=p.decoder, token=os.getenv('HF_TOKEN'))
36
 
37
  text = asr(audio)["text"]
38
  return text
 
32
  # load processor
33
  p = Wav2Vec2ProcessorWithLM.from_pretrained(model_map[language], revision=revision)
34
  # load eval pipeline
35
+ asr = pipeline("automatic-speech-recognition", model=model_map[language], tokenizer=p.tokenizer, feature_extractor=p.feature_extractor, decoder=p.decoder, token=os.getenv('HF_TOKEN'), device=0)
36
 
37
  text = asr(audio)["text"]
38
  return text