Hugging Face's logo

Spaces:
ktangri
/
call-sentiment
Runtime error

App Files Community
ktangri commited on
Commit
1e629a8
•
1 Parent(s): 3e288c9

Changed model to LM

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline, Wav2Vec2ProcessorWithLM
3
  from librosa import load, resample
4
 
5
 
6
- asr_model = 'facebook/wav2vec2-base-960h'
7
  processor = Wav2Vec2ProcessorWithLM.from_pretrained(asr_model)
8
  asr = pipeline('automatic-speech-recognition', model=asr_model, tokenizer=processor.tokenizer, feature_extractor=processor.feature_extractor, decoder=processor.decoder)
9
 
 
3
  from librosa import load, resample
4
 
5
 
6
+ asr_model = 'patrickvonplaten/wav2vec2-base-100h-with-lm'
7
  processor = Wav2Vec2ProcessorWithLM.from_pretrained(asr_model)
8
  asr = pipeline('automatic-speech-recognition', model=asr_model, tokenizer=processor.tokenizer, feature_extractor=processor.feature_extractor, decoder=processor.decoder)
9