yaya-sy commited on
Commit
a7bc14e
·
1 Parent(s): 19f28fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ processor = AutoProcessor.from_pretrained(model_id)
15
  model = Wav2Vec2ForCTC.from_pretrained(model_id)
16
  model = BetterTransformer.transform(model)
17
 
18
- pipe = pipeline("automatic-speech-recognition", model=model)
19
 
20
  def transcribe(audio_file_mic=None, audio_file_upload=None):
21
  if audio_file_mic:
 
15
  model = Wav2Vec2ForCTC.from_pretrained(model_id)
16
  model = BetterTransformer.transform(model)
17
 
18
+ pipe = pipeline("automatic-speech-recognition", model=model, tokenizer=processor.tokenizer, feature_extractor=processor.feature_extractor)
19
 
20
  def transcribe(audio_file_mic=None, audio_file_upload=None):
21
  if audio_file_mic: