mohammed commited on
Commit
e14fb0d
·
verified ·
1 Parent(s): eeda588

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -0
README.md CHANGED
@@ -31,9 +31,16 @@ pip install -U optimum[exporters,onnxruntime] transformers
31
  pip install huggingface_hub
32
 
33
  ```python
 
 
 
 
 
 
34
  from optimum.onnxruntime import ORTModelForSpeechSeq2Seq
35
  from transformers import WhisperTokenizerFast, WhisperFeatureExtractor, pipeline
36
 
 
37
  model_name = 'mohammed/quantized-whisper-small' # folder name
38
  model = ORTModelForSpeechSeq2Seq.from_pretrained(model_name, export=False)
39
  tokenizer = WhisperTokenizerFast.from_pretrained(model_name)
 
31
  pip install huggingface_hub
32
 
33
  ```python
34
+
35
+ # uncomment the following installation if you are using a notebook:
36
+ #!pip install -U optimum[exporters,onnxruntime] transformers
37
+ #!pip install huggingface_hub
38
+
39
+ # import the required packages
40
  from optimum.onnxruntime import ORTModelForSpeechSeq2Seq
41
  from transformers import WhisperTokenizerFast, WhisperFeatureExtractor, pipeline
42
 
43
+ # set model name/id
44
  model_name = 'mohammed/quantized-whisper-small' # folder name
45
  model = ORTModelForSpeechSeq2Seq.from_pretrained(model_name, export=False)
46
  tokenizer = WhisperTokenizerFast.from_pretrained(model_name)