Hugging Face's logo

Spaces:
ktangri
/
call-sentiment
Runtime error

App Files Community
ktangri commited on
Commit
7db1162
•
1 Parent(s): 4f3fae8

Upgrade asr model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from pyannote.audio import Pipeline
4
  from librosa import load, resample
5
  from rpunct import RestorePuncts
6
 
7
- asr_model = 'patrickvonplaten/wav2vec2-base-100h-with-lm'
8
  processor = Wav2Vec2ProcessorWithLM.from_pretrained(asr_model)
9
  asr = pipeline('automatic-speech-recognition', model=asr_model, tokenizer=processor.tokenizer, feature_extractor=processor.feature_extractor, decoder=processor.decoder)
10
  speaker_segmentation = Pipeline.from_pretrained("pyannote/speaker-segmentation")
 
4
  from librosa import load, resample
5
  from rpunct import RestorePuncts
6
 
7
+ asr_model = 'patrickvonplaten/wav2vec2-base-960h-4-gram'
8
  processor = Wav2Vec2ProcessorWithLM.from_pretrained(asr_model)
9
  asr = pipeline('automatic-speech-recognition', model=asr_model, tokenizer=processor.tokenizer, feature_extractor=processor.feature_extractor, decoder=processor.decoder)
10
  speaker_segmentation = Pipeline.from_pretrained("pyannote/speaker-segmentation")