bcci commited on
Commit
2032c87
·
verified ·
1 Parent(s): b5998f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,12 +7,13 @@ from fastapi.responses import HTMLResponse
7
  # Import your model and VAD libraries.
8
  from silero_vad import VADIterator, load_silero_vad
9
 
10
- from transformers import AutoProcessor, pipeline
11
  from optimum.onnxruntime import ORTModelForSpeechSeq2Seq
12
 
13
  processor = AutoProcessor.from_pretrained("onnx-community/whisper-tiny.en")
14
  model = ORTModelForSpeechSeq2Seq.from_pretrained("onnx-community/whisper-tiny.en", subfolder="onnx")
15
- pipe = pipeline("automatic-speech-recognition", model=model, tokenizer=processor.tokenizer, feature_extractor=processor.feature_extractor)
 
16
 
17
  # Constants
18
  SAMPLING_RATE = 16000
 
7
  # Import your model and VAD libraries.
8
  from silero_vad import VADIterator, load_silero_vad
9
 
10
+ from transformers import AutoProcessor, pipeline, WhisperTokenizerFast
11
  from optimum.onnxruntime import ORTModelForSpeechSeq2Seq
12
 
13
  processor = AutoProcessor.from_pretrained("onnx-community/whisper-tiny.en")
14
  model = ORTModelForSpeechSeq2Seq.from_pretrained("onnx-community/whisper-tiny.en", subfolder="onnx")
15
+ tokenizer = WhisperTokenizerFast.from_pretrained("openai/whisper-tiny", language="english")
16
+ pipe = pipeline("automatic-speech-recognition", model=model, tokenizer=tokenizer, feature_extractor=processor.feature_extractor, batch_size=16)
17
 
18
  # Constants
19
  SAMPLING_RATE = 16000