Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|